hihocoder #1041 : 国庆出游(DFS bitset保存子树集合用于指导DFS遍历过程)

题目链接:点击打开链接

题意描述:小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的要求。


解题思路:DFS预处理子树集合(bitset)

分析:根据题意描述,我们可以模拟上述过程小Hi的旅游路线,问题的关键在于DFS过程中我们走那条边呢?

上述问题可以通过DFS预处理以节点i为根节点的子树包含哪些节点,通过bitset保存,这样在DFS过程中我们就可以根据bitset选择走哪条边


代码如下:

/*******************
通过预处理dfs1,求得以节点i为根的树包含哪些节点,使用bitset记录
运用预处理得到的bitset指引dfs2(在dfs2中我们可以根据bitset的内容
按一定的顺序进行遍历)
*******************/
#include <cstdio>
#include <cstring>
#include <bitset>
#define MAXN 105
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int head[MAXN],tol;
struct Edge{
    int v,next;
}e[MAXN*2];
void addEdge(int u,int v){
    e[tol].v=v;e[tol].next=head[u];head[u]=tol++;
    e[tol].v=u;e[tol].next=head[v];head[v]=tol++;
}
int n,m;
bitset<MAXN> bt[MAXN];///bt[i]表示以节点i为根的树包含的那些节点,用于指导dfs2
void dfs1(int u,int p){///预处理,求得bt[maxn]
    bt[u].set(u);
    int v;
    for(int k=head[u];k!=-1;k=e[k].next){
        v=e[k].v;
        if(v!=p){
            dfs1(v,u);
            bt[u]|=bt[v];
        }
    }
}
int go[MAXN];///保存游历顺序
int cnt;///当前走到游历顺序中的第几个位置
bool vis[MAXN];
bool dfs2(int u){
    if(go[cnt]==u) cnt++;
    if(cnt==m) return true;
    vis[u]=true;
    while(cnt<m){
        int tmp=cnt;
        for(int k=head[u];k!=-1;k=e[k].next){
            int v=e[k].v;
            if(bt[v].test(go[cnt])&&!vis[v]){
                vis[v]=true;
                if(dfs2(v)) return true;
            }
        }
        if(cnt==tmp) return false;
    }
    return false;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        tol=0;mem(head,-1);
        scanf("%d",&n);
        int u,v; for(int i=1;i<n;++i){ scanf("%d%d",&u,&v); addEdge(u,v); }
        scanf("%d",&m);
        for(int i=0;i<m;++i) scanf("%d",&go[i]);
        for(int i=1;i<=n;++i) bt[i].reset();
        dfs1(1,-1);
        mem(vis,false);cnt=0;
        if(dfs2(1)) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值