HihoCoder

来源:互联网 发布:wps的数据有效性在哪 编辑:程序博客网 时间:2024/06/06 03:14

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

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


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

AC代码:

#include<bits/stdc++.h>using namespace std;#define maxn 105vector<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;}