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

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<stack>
#include<cmath>
#include<map>
#include<set>
#include<bitset>

#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define MID(x,y) ((x+y)>>1)

#define bug(msg) cout << msg << endl;

using namespace std;
#define eps 1e-8
#define N 105

bitset<N>bt[N];
int n,m;
int aRoad[N];
vector<int>vPath[N];
bool isVisited[N];
int nCurPos;

void clearPath()
{
    for (int i = 0; i <= n; i++)
        vPath[i].clear();
}

void inputPath()
{
    int nFrom, nTo;
    for (int i = 0; i < n - 1; i++)
    {
        scanf("%d%d", &nFrom, &nTo);
        vPath[nFrom].push_back(nTo);
        vPath[nTo].push_back(nFrom);
    }
}

void inputRoad()
{
    scanf("%d", &m);
    for (int i = 0; i < m; i++)
        scanf("%d", &aRoad[i]);
}

void clearBitset()
{
    for (int i = 1; i <= n; i++)
    {
        bt[i].reset();
    }
}

void dfs_initBitset(int nNode, int nPreNode)
{
    bt[nNode].set(nNode);
    for (int i = 0; i < vPath[nNode].size(); i++)
    {

        int nTo = vPath[nNode][i];
        if (nTo == nPreNode)
            continue;

        dfs_initBitset(nTo, nNode);
        bt[nNode] |= bt[nTo];
    }
}

bool dfs(int nNode, int nPreNode)
{

    isVisited[nNode] = true;
    if (nNode == aRoad[nCurPos])
        nCurPos ++;
    if (nCurPos == m)
        return true;
    while (nCurPos < m) //这里面对于兄弟节点，我们不知道应该先处理谁,所以这样
    {
        int nTmpPos = nCurPos;
        for (int i = 0; i < vPath[nNode].size(); i++)
        {
            int nTo = vPath[nNode][i];
            if (nTo == nPreNode)
                continue;

            if (bt[nTo].test(aRoad[nCurPos]) && !isVisited[nTo])
            {
                if (dfs(nTo, nNode))
                    return true;
            }
        }
        if (nCurPos == nTmpPos)
            return false;
    }
    return false;
}

bool solve()
{
    clearBitset();
    dfs_initBitset(1, -1);
    memset(isVisited, false, sizeof(isVisited));
    nCurPos = 0;
    return dfs(1, -1);
}

int main()
{
    int nCase = 0;
    scanf("%d", &nCase);
    while (nCase --)
    {
        scanf("%d", &n);
        clearPath();
        inputPath();
        inputRoad();
        if (solve())
            cout << "YES" << endl;
        else
            cout << "NO" << endl;
    }
    return 0;
}