hihocoder #1041 : 国庆出游 （DFS）

#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

题解：给你一个强连通图和一条规定的路，问你存不存在这条一条路，注意路是可以来回走一遍的。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int m,ok,j;
int order[N];
int flag[N][N];
bitset<N>digit[N];
vector<int>road[N];

void Check_Reach(int a, int v)  //是否能到达 
{  
    digit[a][a] = 1;  
    for(int i = 0; i < road[a].size(); i++)  
    {  
        int b = road[a][i];  
        if(b == v)  
            continue;  
        Check_Reach(b, a);  
        digit[a] |= digit[b];  //按位或 
    }  
}  
  
void dfs(int a, int v)  
{  
    if(a == order[j])  
        j++;  
    if(j == m)  //到达终点 
    {  
        ok = 1;  
        return;  
    }  
    while(j < m)  
    {  
        int p = j, c = order[j];  
        for(int i = 0; i < road[a].size(); i++)  
        {  
            int b = road[a][i];  
            if(b == v)  
                continue;  
            if(digit[b][c] && flag[a][b])  
            {  
                flag[a][b] = 0;  
                dfs(b, a);  
                break;  
            }  
        }  
        if(p == j)  
            break;  
    }  
}  

int main()
{
	int a,b,t,n;
	scanf("%d",&t);
	while(t--)
	{
		ok=j=0;
		memset(flag,0,sizeof(flag));
		for(int i=0;i<N;i++)
		{
			digit[i].reset();
			road[i].clear();
		}
		scanf("%d", &n);  
        for(int i = 1; i < n; i++)  
        {  
            scanf("%d%d", &a, &b);  
            road[a].push_back(b);  
            road[b].push_back(a);  
            flag[a][b] = flag[b][a] = 1;  
        }  
        scanf("%d", &m);  
        for(int i = 0; i < m; i++)  
            scanf("%d", &order[i]);  
        Check_Reach(1, -1);  
        dfs(1, -1);  
        if(ok)  
            printf("YES\n");  
        else  
            printf("NO\n");  
	}
	return 0;
}