B - 国庆出游 HihoCoder - 1041

大意是每条边走2遍,并且是否按给出的顺序访问某些城市,输出yes或no。

一开始写了个TLE的傻逼搜索把每条边都走了两遍。

关键在于,这是一棵树!2个点之间的路径唯一。还有一个重要的思想就是从1个点到另一个点时,其他不在那个顺序里的点就当顺路走掉了,不用考虑。

于是就随那个顺序遍历就行。

 

#include
#include
#include
#include
#define maxl 101

using namespace std;

int n,t,m,cnt=0;
int a[maxl],dep[maxl],fa[maxl];
vector  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;idep[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;
}

 

 

 

 

 

你可能感兴趣的:(dfs)