hihocoder——1041国庆出游（搜索）

来源：互联网发布：sql语言培训班编辑：程序博客网时间：2024/05/16 19:40

描述

小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的要求。

样例输入

271 21 32 42 53 63 733 7 271 21 32 42 53 63 733 2 7

样例输出

YESNO

巧妙的利用了二进制记录了每个节点的所有子节点。

#include<cstdio>#include<cstdlib>#include<cstring>#include<vector>#include<bitset>using namespace std;#define MAX 111int m, flag, tmp, order[MAX], tmap[MAX][MAX];bitset<MAX> digit[MAX];vector<int> q[MAX];void work(int a, int v)         //确定每个根节点下面有哪些节点{    digit[a][a] = 1;    for(int i = 0; i < q[a].size(); i++)    {        int b = q[a][i];        if(b == v)            continue;        work(b, a);        digit[a] |= digit[b];    //巧妙的运用二进制实现    }}void dfs(int a, int v){    if(a == order[tmp])        tmp++;    if(tmp == m)  //如果已经走过了特定的m个点，返回    {        flag = 1;        return;    }    while(tmp < m)    {        int p = tmp;        int c = order[tmp];        for(int i = 0; i < q[a].size(); i++)        {            int b = q[a][i];            if(b == v)                continue;            if(digit[b][c] && tmap[a][b])      // 如果b下面有c  而且a b联通            {                tmap[a][b] = 0;             //截断，防止重复走                dfs(b, a);                break;            }        }        if(p == tmp)            break;    }}int main(void){    int a, b, i, t, n;    scanf("%d", &t);    while(t--)    {        flag = tmp = 0;        memset(tmap, 0, sizeof (tmap));        for(i = 0; i < MAX; i++)        {            digit[i].reset();            q[i].clear();        }        scanf("%d", &n);        for(i = 1; i < n; i++)        {            scanf("%d%d", &a, &b);            q[a].push_back(b);            q[b].push_back(a);            tmap[a][b] = tmap[b][a] = 1;        }        scanf("%d", &m);        for(i = 0; i < m; i++)            scanf("%d", &order[i]);        work(1, -1);        dfs(1, -1);        if(flag)            printf("YES\n");        else            printf("NO\n");    }}

0 0