描述
小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;
typedef long long ll;
#define maxn 105
struct line{
int V; //顶点编号
int w; //此顶点的访问次数,最多访问两次
line *next;
line(){ //构造函数初始化
V = w = 0;
next = NULL;
}
};
bool f[maxn]; //一个标记数组,后面会用到
line adj[maxn]; //创建一个邻接链表
void init(){ //初始化邻接链表
for(int i = 0; i < maxn; i++) {
adj[i].V = i;
adj[i].next = NULL;
}
}
void add(int u,int v) {//在链表中添加
line *head = &adj[u];
line *p = new line;
p->V = v;
p->next = head->next;
head->next = p;
}
int flag = 0; //flag为最后判断yes或no的标准
void dfs(line *u,line *v) {//深度优先搜索,搜索从u到v的可能性
flag = 0;
if(u->V == v->V ) {//若当前节点编号等于目标节点编号,则能到达
flag = 1;
return;
}
if(f[u->V]) return;//因为有特定访问顺序,如3,7,2,
//不能在7前面就访问2
u->w++;//已访问当前,则权加一
line *p;//对当前节点的所有可能到达的下一个节点递归搜索
for(p = adj[u->V].next; p != NULL; p = p->next) {
if(p->w < 2) dfs(p,v);
if(flag) break;//若果已经找到可行,则不需要访问其他节点了
}
}
int main(){
ios::sync_with_stdio(false);
int u,v;
int t,n,m,like[maxn];
cin >> t;
while(t--) {
init();
cin >> n;
for(int i = 1; i < n; i++) {
cin >> u >> v;
add(u,v);
add(v,u);
}
//下面注释掉得代码可以检验邻接表的正确性
/* for(int i=1; i<=n; ++i){
line * p;
cout << "adj[" << i <<"]";
for(p = adj[i].next; p != NULL; p = p->next)
cout << " <- " << p->V ;
cout << endl;
} */
cin >> m;
memset(f,0,sizeof(f));
like[0] = 1;//like[]储存题目中所给的序列,隐含从1号开始
for(int i = 1; i <= m; i++) {
cin >> like[i];
f[like[i]] = 1;//先对这些节点标记为1
}
//每次都两个两个的dfs
for(int i = 0; i <= m-1; i++)
{
f[like[i]] = 0;//当前要操作的两个节点,标记改为0
f[like[i+1]] = 0;
dfs(&adj[like[i]],&adj[like[i+1]]);//这要有&符
if(!flag) break;
}
if(flag) cout << "YES" << endl;
else cout << "NO" << endl;
}
return 0;
}