描述
小Hi和小Ho准备国庆期间去A国旅游。A国的城际交通比较有特色:它共有n座城市(编号1-n);城市之间恰好有n-1条公路相连,形成一个树形公路网。小Hi计划从A国首都(1号城市)出发,自驾遍历所有城市,并且经过每一条公路恰好两次——来回各一次——这样公路两旁的景色都不会错过。
令小Hi苦恼的是他的小伙伴小Ho希望能以某种特定的顺序游历其中m个城市。例如按3-2-5的顺序游历这3座城市。(具体来讲是要求:第一次到达3号城市比第一次到达2号城市早,并且第一次到达2号城市比第一次到达5号城市早)。
小Hi想知道是否有一种自驾顺序满足小Ho的要求。
输入
输入第一行是一个整数T(1<=T<=20),代表测试数据的数量。
每组数据第一行是一个整数n(1 <= n <= 100),代表城市数目。
之后n-1行每行两个整数a和b (1 <= a, b <= n),表示ab之间有公路相连。
之后一行包含一个整数m (1 <= m <= n)
最后一行包含m个整数,表示小Ho希望的游历顺序。
输出
YES或者NO,表示是否有一种自驾顺序满足小Ho的要求。
样例输入
2
7
1 2
1 3
2 4
2 5
3 6
3 7
3
3 7 2
7
1 2
1 3
2 4
2 5
3 6
3 7
3
3 2 7
样例输出
YES
NO
巧妙的利用了二进制记录了每个节点的所有子节点。
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<bitset>
using namespace std;
#define MAX 111 int m, flag, tmp, order[MAX], tmap[MAX][MAX];
bitset<MAX> digit[MAX];
vector<int> q[MAX]; void work(int a, int v) //确定每个根节点下面有哪些节点
{
digit[a][a] = 1;
for(int i = 0; i < q[a].size(); i++)
{
int b = q[a][i];
if(b == v)
continue;
work(b, a);
digit[a] |= digit[b]; //巧妙的运用二进制实现
}
} void dfs(int a, int v)
{
if(a == order[tmp])
tmp++;
if(tmp == m) //如果已经走过了特定的m个点,返回
{
flag = 1;
return;
}
while(tmp < m)
{
int p = tmp;
int c = order[tmp];
for(int i = 0; i < q[a].size(); i++)
{
int b = q[a][i];
if(b == v)
continue;
if(digit[b][c] && tmap[a][b]) // 如果b下面有c 而且a b联通
{
tmap[a][b] = 0; //截断,防止重复走
dfs(b, a);
break;
}
}
if(p == tmp)
break;
}
} int main(void)
{
int a, b, i, t, n;
scanf("%d", &t);
while(t--)
{
flag = tmp = 0;
memset(tmap, 0, sizeof (tmap));
for(i = 0; i < MAX; i++)
{
digit[i].reset();
q[i].clear();
}
scanf("%d", &n);
for(i = 1; i < n; i++)
{
scanf("%d%d", &a, &b);
q[a].push_back(b);
q[b].push_back(a);
tmap[a][b] = tmap[b][a] = 1;
}
scanf("%d", &m);
for(i = 0; i < m; i++)
scanf("%d", &order[i]);
work(1, -1);
dfs(1, -1);
if(flag)
printf("YES\n");
else
printf("NO\n");
}
}