hihocoder #1041 : 国庆出游

#1041 : 国庆出游

时间限制:1000ms
单点时限:1000ms
内存限制:256MB

描述

小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

核心思想：在指定的游览序列中，任意的节点n，其所有子孙节点必须紧跟其后。（紧跟！）（该节点下面的所有节点）

如：n的子孙节点包括abc，那么nabc、nacb都是合法的

na*bc、n*abc都是不合法的

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<vector>

#include<bitset>

using namespace std;

#define MAX 111

int m, ok, j, order[MAX], flag[MAX][MAX];

bitset<MAX> digit[MAX];//存放每个城市能到达的所有城市，能到达不是两城市间必须有路

vector<int> u[MAX];//存放每个城市有路连接的所有城市

void canReach(int a, int v)

{

digit[a][a] = 1;

for(int i = 0; i < u[a].size(); i++)

{

int b = u[a][i];

if(b == v)

continue;

canReach(b, a);

digit[a] |= digit[b];

}

}

void dfs(int a, int v)

{

if(a == order[j])

j++;

if(j == m)

{

ok = 1;

return;

}

while(j < m)

{

int p = j, c = order[j];

for(int i = 0; i < u[a].size(); i++)

{

int b = u[a][i];

if(b == v)

continue;

if(digit[b][c] && flag[a][b])//如果b能到达c并且a、b之间有路

{

flag[a][b] = 0;

dfs(b, a);

break;

}

}

if(p == j)

break;//表示以该节点为根的子树不存在目标点，跳出循环，（然后返回递归上一层）

}

}

int main(void)

{

int a, b, i, t, n;

scanf("%d", &t);

while(t--)

{

ok = j = 0;

memset(flag, 0, sizeof flag);

for(i = 0; i < MAX; i++)

{

digit[i].reset();

u[i].clear();

}

scanf("%d", &n);

for(i = 1; i < n; i++)

{

scanf("%d%d", &a, &b);

u[a].push_back(b);

u[b].push_back(a);

flag[a][b] = flag[b][a] = 1;

}

scanf("%d", &m);

for(i = 0; i < m; i++)

scanf("%d", &order[i]);

canReach(1, -1);

dfs(1, -1);

if(ok)

printf("YES\n");

else

printf("NO\n");

}

}