hihocoder 1041 國慶出遊 (DFS + bitset 好題)

_TCgogogo_發表於2016-03-19
原文網址 : https://blog.csdn.net/tc_to_top/article/details/50931608
時間限制: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

題目連結：http://hihocoder.com/problemset/problem/1041


題目分析：用bitset存一下當前結點往下能走到的結點，然後搜尋的時候判斷一下下一個要走的點在不在當前結點下面的路徑下，相當於模擬，如果有一次往下搜的時候找不到該走的下一個點則說明遇到死路不可能完成遍歷

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <bitset>
using namespace std;
int const MAX = 205;
int head[MAX], od[MAX], cnt, n, m, num;
bool vis[MAX], flag;
bitset <MAX> sta[MAX];

struct EDGE
{
    int to, nxt;
}e[MAX];

void Init()
{
    for(int i = 0; i < MAX; i++)
        sta[i].reset();
    memset(head, -1, sizeof(head));
    cnt = 0;
}

void Add(int u, int v)
{
    e[cnt].to = v;
    e[cnt].nxt = head[u];
    head[u] = cnt ++;
}

void DFS(int u, int fa)
{
    sta[u][u] = 1;
    vis[u] = true;
    for(int i = head[u]; i != -1; i = e[i].nxt)
    {
        int v = e[i].to;
        if(!vis[v] && v != fa) 
        {
            DFS(v, u);
            sta[u] |= sta[v];
        }
    }
}

void Solve(int u, int fa)
{
    if(flag)
        return;
    if(u == od[num])
        num ++;
    if(num == m)
    {
        flag = true;
        return;
    }
    vis[u] = true;
    while(num < m)
    {
        int tmp = num;
        for(int i = head[u]; i != -1; i = e[i].nxt)
        {
            int v = e[i].to;
            if(v != fa && sta[v][od[num]] && !vis[v])
                Solve(v, u);
        }
        if(tmp == num)
            break;
    }
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T --)
    {
        Init();
        scanf("%d", &n);
        int u, v;
        for(int i = 0; i < n - 1; i++)
        {
            scanf("%d %d", &u, &v);
            Add(u, v);
            Add(v, u);
        }
        scanf("%d", &m);
        for(int i = 0; i < m; i++)
            scanf("%d", &od[i]);
        memset(vis, false, sizeof(vis));
        DFS(1, -1);
        flag = false;
        num = 0;
        memset(vis, false, sizeof(vis));
        Solve(1, -1);
        printf("%s\n", flag ? "YES" : "NO");
    }
}


相關文章