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;
}