大意是每条边走2遍,并且是否按给出的顺序访问某些城市,输出yes或no。
一开始写了个TLE的傻逼搜索把每条边都走了两遍。
关键在于,这是一棵树!2个点之间的路径唯一。还有一个重要的思想就是从1个点到另一个点时,其他不在那个顺序里的点就当顺路走掉了,不用考虑。
于是就随那个顺序遍历就行。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define maxl 101
using namespace std;
int n,t,m,cnt=0;
int a[maxl],dep[maxl],fa[maxl];
vector <int> e[maxl],tree[maxl];
bool yes;
bool vis[maxl];
int num[maxl];
void prework()
{
int u,v;
scanf("%d",&n);
for(int i=1;i<=n;i++)
e[i].clear(),tree[i].clear();
for(int i=1;i<=n-1;i++)
{
scanf("%d%d",&u,&v);
e[u].push_back(v);
e[v].push_back(u);
}
yes=true;
memset(vis,false,sizeof(vis));
scanf("%d",&m);
for(int i=1;i<=m;i++)
scanf("%d",&a[i]);
}
void dfs(int u,int depth)
{
dep[u]=depth;vis[u]=true;
int v,l=e[u].size();
for(int i=0;i<l;i++)
{
v=e[u][i];
if(!vis[v])
fa[v]=u,dfs(v,depth+1);
}
}
void mainwork()
{
dep[1]=0;
dfs(1,1);
int u,v;a[0]=1;
memset(num,0,sizeof(num));
for(int i=1;i<=m;i++)
{
u=a[i-1];v=a[i];
if(dep[u]<dep[v])
swap(u,v);
while(dep[u]>dep[v])
{
if(num[u]==2)
{
yes=false;
return;
}
num[u]++;
u=fa[u];
}
while(u!=v)
{
if(num[u]==2 || num[v]==2)
{
yes=false;
return;
}
num[u]++;num[v]++;
u=fa[u];v=fa[v];
}
}
}
void print()
{
if(yes)
printf("YES\n");
else
printf("NO\n");
}
int main()
{
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
prework();
mainwork();
print();
}
return 0;
}