#1041 : 国庆出游
时间限制:1000ms
单点时限:1000ms
内存限制:256MB
- 样例输入
-
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
描述
小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的要求。
思路:1为起点,bitset容器保存以某节点为根节点,子树能到达的所有节点,然后从第一个顺序点找起,dfs到达过程将走过的路都“消掉”,表示“回程”时已经走过了,这样搜索下一个顺序点时就不会再走这些路,从而能判断能否走完所有顺序点。
# include <iostream> # include <cstdio> # include <cstring> # include <vector> # include <bitset> # include <algorithm> # define MAXN 103 using namespace std; bitset<MAXN>f[MAXN]; vector<int>v[MAXN]; int a[MAXN], r[MAXN][MAXN], cnt, flag, t, n, m;; void son_tree(int cur, int pre)//每一个子树能到达的点 { f[cur][cur] = 1; for(int i=0; i<v[cur].size(); ++i) { int dot = v[cur][i]; if(dot == pre) continue; son_tree(dot, cur); f[cur] |= f[dot]; } } void dfs(int cur, int pre) { if(cnt < m && cur == a[cnt]) ++cnt; if(cnt == m) { flag = 1; return; } while(cnt < m)//cnt是指进行的顺序点的数组下标 { int next = a[cnt]; int now = cnt; for(int i=0; i<v[cur].size(); ++i) { int dot = v[cur][i]; if(dot == pre) continue; if(f[dot][next] && r[cur][dot])//该子树能到达目标点且路径通畅(以前没走过)。 { r[cur][dot] = 0; dfs(dot, cur); break; } } if(cnt == now) break; } } int main() { scanf("%d",&t); while(t--) { scanf("%d",&n); memset(r, 0, sizeof(r)); for(int i=1; i<=n; ++i) { v[i].clear(); f[i].reset(); } for(int i=0; i<n-1; ++i) { int x, y; scanf("%d%d",&x,&y); r[x][y] = r[y][x] = 1; v[x].push_back(y); v[y].push_back(x); } scanf("%d",&m); for(int i=0; i<m; ++i) scanf("%d",&a[i]); cnt = flag = 0; son_tree(1, -1); dfs(1, -1); if(flag) puts("YES"); else puts("NO"); } return 0; }