codeforces 529div3 problem D

题意是给你一个有向环,让你根据给你的信息求出这个环的序列,给你的信息是这个节点的下面的两个节点

思路:首先应该想到是搜索的方法,在深搜过程中,传递一个当前的节点,和祖宗节点,祖宗节点的作用是确保当前节点的下一个节点也在祖宗节点的后连个节点中,然后就是确保有向,这个我用的是path数组,来维护有向关系,如果dfs返回true,就标记当前孩子节点是由当前节点传递而来。

代码:(如果被hack,我再改)

#include
#include
#include
#include

using namespace std;

const int maxn=2e5+10;
struct child
{
	int ch[2];//记录这个节点的两个孩子节点
}child[maxn];
int used[maxn];
int path[maxn];
int flag=0;
int cnt=1,n;

bool dfs(int p,int root)
{
	if(p==flag&&cnt==n)	
	{
		return true;
	}
	for(int i=0;i<2;i++)
	{
		if(used[child[p].ch[i]])	continue;
		//used[p.ch[i]]=1;
		if(child[p].ch[i]!=child[root].ch[0]&&child[p].ch[i]!=child[root].ch[1])	continue;//如果孩子不是祖宗节点的孙子节点,则代表不合法
		used[child[p].ch[i]]=1;
		cnt++;//用来计算是否遍历所有的节点,避免中间搜索就返回的情况
		if(dfs(child[p].ch[i],p))
		{
			//printf("%d ",p);
			path[child[p].ch[i]]=p;
			return true;
		}
		used[child[p].ch[i]]=0;
		cnt--;
	}
	return false;
}

int main()
{
	//int n;
	scanf("%d",&n);
	memset(path,-1,sizeof(path));
	for(int i=1;i<=n;i++)
		scanf("%d %d",&child[i].ch[0],&child[i].ch[1]);
	flag=1;
	for(int i=0;i<2;i++)
	{
		used[child[1].ch[i]]=1;
		if(dfs(child[1].ch[i],1))
		{
			//printf("1\n");
			path[child[1].ch[i]]=1;
			break;
		}
		used[child[1].ch[i]]=0;
	}
	stackiron;
	iron.push(1);
	int p=1;
	while(path[p]!=flag)
	{
		iron.push(path[p]);
		p=path[p];
	}
	while(!iron.empty())
	{
		printf("%d ",iron.top());
		iron.pop();
	}
	printf("\n");
	//system("pause");
	return 0;
}

 

你可能感兴趣的:(搜索)