HihoCoder - 1041: 国庆出游(dfs、邻接表)

2 篇文章 0 订阅
2 篇文章 0 订阅

题目链接:HihoCoder - 1041: 国庆出游

题目大意:已知n个城市之间有n-1条路,现在给定部分城市的旅游顺序,问能否刚好每条路都走两次,且按给定顺序参观完成旅行


思路:经典dfs搜索,先用bitset计算每个节点的子节点(按位或运算加速),定义vector存邻接表,并用一个二维数组存各城市的连接情况,即邻接矩阵,然后将给定的旅游顺序存到一个一维数组,用这个顺序进行dfs搜索,如果能按这个顺序走完m个城市,则说明计划可行(其余的节点不用管它,总是有办法走完的)

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define maxn 105
vector<int>adj[maxn];//定义邻接表
int vis[maxn];//访问顺序
bitset<maxn>child[maxn];//计算每个节点对应的子节点
int graph[maxn][maxn];//定义邻接矩阵

void cntchild(int son,int par)//注意:有子节点并不代表有路直接相连,可能有中转
{
    child[son][son]=1;
    for(int i=0; i<(int)adj[son].size(); i++)//遍历子节点下的孙节点
    {
        int grandson=adj[son][i];
        if(grandson==par) continue;//防止回路,不走父节点
        cntchild(grandson,son);//循环
        child[son]|=child[grandson];//利用或运算加速判断子节点下的孙节点
    }
}

bool dfs(int v[],int &id,int m,int son,int par)
{
    if(v[id]==son && id<m) id++;
    if(id==m) return true;
    while(id<m)
    {
        int node=vis[id];
        int newid=id;
        for(int i=0; i<(int)adj[son].size(); i++)
        {
            int grandson=adj[son][i];
            if(grandson==par) continue;
            if(child[grandson][node] && graph[son][grandson])//该节点下的子节点包含目标节点,即可以到达
            {
             graph[son][grandson]=0;//拆桥,保证只走一次
                if(dfs(v,id,m,grandson,son)) return true;//深度遍历查找
            }
        }
        if(newid==id) break;//防止陷入无限循环,没有答案时退出
    }
    return false;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
        {
            adj[i].clear();
            child[i].reset();
        }
        memset(graph,0,sizeof(graph));
        int a,b;
        for(int i=0; i<n-1; i++)
        {
            scanf("%d%d",&a,&b);
            adj[a].push_back(b);
            adj[b].push_back(a);
            graph[a][b]=graph[b][a]=1;
        }
        int m;
        scanf("%d",&m);
        for(int i=0; i<m; i++)
            scanf("%d",&vis[i]);
        cntchild(1,-1);
        int id=0;
        if(dfs(vis,id,m,1,-1)) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值