hihoCoder1041：国庆出游（bitset引导dfs遍历）

#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

题意：一棵无回路树，以1为起点，判断能否按顺序遍历指定的点，且每条边只能走两遍。

思路：1为起点，bitset容器保存以某节点为根节点，子树能到达的所有节点，然后从第一个顺序点找起，dfs到达过程将走过的路都“消掉”，表示“回程”时已经走过了，这样搜索下一个顺序点时就不会再走这些路，从而能判断能否走完所有顺序点。

# include <iostream>
# include <cstdio>
# include <cstring>
# include <vector>
# include <bitset>
# include <algorithm>
# define MAXN 103
using namespace std;
bitset<MAXN>f[MAXN];
vector<int>v[MAXN];
int a[MAXN], r[MAXN][MAXN], cnt, flag, t, n, m;;

void son_tree(int cur, int pre)//每一个子树能到达的点
{
f[cur][cur] = 1;
for(int i=0; i<v[cur].size(); ++i)
{
int dot = v[cur][i];
if(dot == pre) continue;
son_tree(dot, cur);
f[cur] |= f[dot];
}
}

void dfs(int cur, int pre)
{
if(cnt < m && cur == a[cnt]) ++cnt;
if(cnt == m)
{
flag = 1;
return;
}
while(cnt < m)//cnt是指进行的顺序点的数组下标
{
int next = a[cnt];
int now = cnt;
for(int i=0; i<v[cur].size(); ++i)
{
int dot = v[cur][i];
if(dot == pre) continue;
if(f[dot][next] && r[cur][dot])//该子树能到达目标点且路径通畅（以前没走过）。
{
r[cur][dot] = 0;
dfs(dot, cur);
break;
}
}
if(cnt == now) break;
}
}
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(r, 0, sizeof(r));
for(int i=1; i<=n; ++i)
{
v[i].clear();
f[i].reset();
}
for(int i=0; i<n-1; ++i)
{
int x, y;
scanf("%d%d",&x,&y);
r[x][y] = r[y][x] = 1;
v[x].push_back(y);
v[y].push_back(x);
}
scanf("%d",&m);
for(int i=0; i<m; ++i)
scanf("%d",&a[i]);
cnt = flag = 0;
son_tree(1, -1);
dfs(1, -1);
if(flag)
puts("YES");
else
puts("NO");
}
return 0;
}