题目链接: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;
}