hihoCoder--1041 国庆出游(dfs + bitset)

描述
小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 <bits/stdc++.h>
using namespace std;

const int maxn = 105;
int n, m, t;
vector<int> G[maxn];
int a[maxn];
bitset<maxn> hasChild[maxn];    // hasChild[u]表示从当前结点u能往下走的结点集合
bool connect[maxn][maxn];

void initChild(int u = 1, int parent = -1){
    hasChild[u][u] = 1;
    for(int i = 0; i < G[u].size(); ++i){
        int v = G[u][i];
        if(v != parent){
            initChild(v, u);
            hasChild[u] |= hasChild[v];
        }
    }
}

bool dfs(int& idx, int u = 1, int parent = -1){
    if(idx < m && a[idx] == u) idx++;
    if(idx == m) return true;

    while(idx < m){
        int nextNode = a[idx];
        int curidx = idx;
        for(int i = 0; i < G[u].size(); ++i){
            int v = G[u][i];
            if(v != parent && hasChild[v][nextNode] && connect[u][v]){
                connect[u][v] = 0;
                if(dfs(idx, v, u)) return true;
            }
        }
        if(idx == curidx) break;
    }
    return false;
}

int main(){
#ifdef EXMY
freopen("data.in", "r", stdin);
#endif // EXMY

    cin >> t;
    while(t--){
        memset(connect, 0, sizeof(connect));
        for(int i = 0; i <= n; ++i){
            hasChild[i].reset();
            G[i].clear();
        }
        cin >> n;
        for(int i = 0; i < n - 1; ++i){
            int a, b; cin >> a >> b;
            G[a].push_back(b);
            G[b].push_back(a);
            connect[a][b] = connect[b][a] = true;
        }
        cin >> m;
        for(int i = 0; i < m; ++i){
            cin >> a[i];
        }
        initChild();

        int idx = 0;
        if(dfs(idx)) cout << "YES" << endl;
        else cout << "NO" << endl;
    }

    return 0;
}