#1041 : 國慶出遊(dfs+bitset大法)

bitset用法:https://blog.csdn.net/Kuguotao/article/details/89327059

時間限制: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
/*

*/
#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
#include <bitset>
#include <stack>
#define ll long long
#define ull unsigned long long
#define inf 0x3f3f3f3f
const int mod=997;
const int N=110;
using namespace std;
vector<int>ve[N];
bitset<N>bit[N];
int w[N],vis[N][N];
int n,m,p,flag;

void tao(int a,int u)//處理每個節點下面有哪些節點
{
    bit[a][a]=1;
    for(int i=0;i<ve[a].size();i++)
    {
        int v=ve[a][i];
        if(v==u)
            continue;
        tao(v,a);
        cout<<"bit["<<a<<"]["<<bit[a]<<endl<<"bit["<<v<<"]["<<bit[v]<<endl<<"bit["<<a<<"]["<<v<<"]:"<<bit[a][v]<<endl;
        bit[a]|=bit[v];//或全爲零才爲零,
        cout<<"bit["<<a<<"]["<<bit[a]<<endl<<"bit["<<v<<"]["<<bit[v]<<endl<<"bit["<<a<<"]["<<v<<"]:"<<bit[a][v]<<endl<<"bit["<<v<<"]["<<a<<"]:"<<bit[v][a]<<endl;
    }
}
void dfs(int x,int u)
{
    if(x==w[p])
        p++;
    if(p==m)//如果走過了特定的m個點,返回
    {
        flag=1;
        return ;
    }
    while(p<m)
    {
        int pp=p;
        int c=w[p];
        for(int i=0;i<ve[x].size();i++)
        {
            int v=ve[x][i];
            if(v==u)
                continue;
            if(bit[v][c]&&vis[x][v])
            {//如果v下面有c,且x,v聯通
                vis[x][v]=0;//截斷x,v
                dfs(v,x);
                break;
            }
        }
        if(pp==p)
            break;
    }
}
int main()
{

    int t;
    scanf("%d",&t);
    while(t--)
    {
        p=0;
        flag=0;
        scanf("%d",&n);
        memset(vis,0,sizeof(vis));
        for(int i=0;i<N;i++)
        {
            ve[i].clear();
            bit[i].reset();
        }
        for(int i=1;i<n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            ve[a].push_back(b);
            ve[b].push_back(a);
            vis[a][b]=1;vis[b][a]=1;
        }
        scanf("%d",&m);
        for(int i=0;i<m;i++)
            scanf("%d",&w[i]);
         tao(1,-1);
         for(int i=1;i<=n;i++)
            cout<<i<<" "<<bit[i]<<endl;
//         dfs(1,-1);
//         if(flag)
//            printf("YES\n");
//         else
//            printf("NO\n");
    }
}
2
7
1 2
1 3
2 4
2 5
3 6
3 7
3
3 7 2
bit[2][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000100
bit[4][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010000
bit[2][4]:0
bit[2][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010100
bit[4][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010000
bit[2][4]:1
bit[4][2]:0
bit[2][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010100
bit[5][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000100000
bit[2][5]:0
bit[2][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000110100
bit[5][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000100000
bit[2][5]:1
bit[5][2]:0
bit[1][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010
bit[2][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000110100
bit[1][2]:0
bit[1][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000110110
bit[2][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000110100
bit[1][2]:1
bit[2][1]:0
bit[3][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001000
bit[6][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001000000
bit[3][6]:0
bit[3][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001001000
bit[6][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001000000
bit[3][6]:1
bit[6][3]:0
bit[3][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001001000
bit[7][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010000000
bit[3][7]:0
bit[3][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011001000
bit[7][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010000000
bit[3][7]:1
bit[7][3]:0
bit[1][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000110110
bit[3][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011001000
bit[1][3]:0
bit[1][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011111110
bit[3][00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011001000
bit[1][3]:1
bit[3][1]:0
1 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011111110
2 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000110100
3 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011001000
4 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010000
5 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000100000
6 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001000000
7 00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000010000000

 bitset[1]就是倒數第二位爲1,其餘爲零,bitset[2]就是倒數第三位爲1,其餘爲零,以此類推。利用進制與運算表明死否有某一子節點,上圖是樣例數據輸出。

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章