hihocoder 1041 国庆出游 (DFS + bitset 好题)

最新推荐文章于 2022-03-13 16:38:23 发布

_TCgogogo_

最新推荐文章于 2022-03-13 16:38:23 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构搜索文章标签： hihocoder DFS bitset

本文链接：https://blog.youkuaiyun.com/Tc_To_Top/article/details/50931608

ACM 同时被 3 个专栏收录

635 篇文章

订阅专栏

数据结构

129 篇文章

订阅专栏

搜索

66 篇文章

订阅专栏

本文探讨了在树形结构中实现路径遍历并满足特定条件的方法，通过实例解析了如何解决从首都出发，自驾遍历所有城市，且满足特定伙伴游历顺序的问题。利用树的性质和深度优先搜索算法，实现路径的有效规划。

时间限制: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的要求。

样例输入

样例输出

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");
    }
}