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

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）。

深搜加一个剪枝即可：如果当前访问的点要求被第xx个访问，然而并非如此，返回。

至于要求条边都访问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();
}
}
}