日记★枚举★B-国庆出游

B-国庆出游

题目

时间限制: 1 Sec 内存限制: 262144 kB

题目描述

小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

分析

存下每个点应该被第几个访问（没要求则为0）。
深搜加一个剪枝即可：如果当前访问的点要求被第 x x <script type="math/tex" id="MathJax-Element-47">x</script>个访问，然而并非如此，返回。
至于要求条边都访问2次，只要能够遍历一遍，按原路返回即可达成要求。

代码

实现有点困难。

#include<vector>
#include<cstdio>
#include<algorithm>
using namespace std;

int read(){
    int x=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    return x*f;
}

#define MAXN 100
int M;
bool Ans;
int ord[MAXN+5],cnt;
//ord[i]表示i号点要求第几个访问
//cnt表示已经成功地访问了几个点
vector<int> G[MAXN+5];//图

void dfs(int u,int fa){
    if(Ans) return;
    if(ord[u]){
        if(ord[u]!=cnt) return;//不该在这个时间访问这个点
        else{
            cnt++;
            if(cnt>M){//只要能按要求的顺序把有要求的点访问一边，就能完成旅行
                Ans=1;
                return;
            }
        }
    }
    for(int i=0;i<int(G[u].size());i++)
        if(G[u][i]!=fa)
            dfs(G[u][i],u);
    for(int i=G[u].size()-1;i>=0;i--)//正反找两遍避免儿子间次序不同的差异
        if(G[u][i]!=fa)
            dfs(G[u][i],u);
}

int main(){
    int T=read();
    while(T--){
        cnt=1,Ans=0;
        int N=read();
        for(int i=1;i<N;i++){
            int u=read(),v=read();
            G[u].push_back(v);
            G[v].push_back(u);
        }
        M=read();
        for(int i=1;i<=M;i++)
            ord[read()]=i;
        dfs(1,-1);
        if(Ans) puts("YES");
        else puts("NO");
        for(int i=1;i<=N;i++){
            ord[i]=0;
            G[i].clear();
        }
    }
}