Misha and Forest(拓扑排序)

题意:给出一棵树每个节点的相邻节点个数a和相邻节点异或和b。求数的边数和每条边的两个节点。

思路:
应用拓扑排序的思路,由于是棵树所以一定有只和一个点相邻的点,这样的点a[i]是1,b[i]是和这个点相邻的点。在改变点b[i]的a[b[i]]和b[b[i]]

代码:

#include
#include
#include
#include
#include
using namespace std;
const int N=(1<<16)+100;
int a[N],b[N],st[N];
struct Node{
	int x,y;
}edge[N];
int tot;
int main()
{
	int n;
	scanf("%d",&n);
	queue<int>q;
	for(int i=0;i<n;i++)
	{
		scanf("%d%d",&a[i],&b[i]);
	}
	for(int i=0;i<n;i++)
	{
		if(a[i]==1&&st[i]==0)
		{
			q.push(i);
			st[i]=1;
		}
	}
	while(q.size())
	{
		int t=q.front();q.pop();
		if(a[t]!=1) continue;//加入队列的点的值还有可能被改变,因为这个是无向图(拓扑排序时是有向图)
		edge[++tot].x=t;edge[tot].y=b[t];
		a[b[t]]--;
		b[b[t]]^=t;		
		if(a[b[t]]==1&&st[b[t]]==0)
		{
			q.push(b[t]);
			st[b[t]]=1;
		}

	}
	printf("%d\n",tot);
	for(int i=1;i<=tot;i++)
	{
		printf("%d %d\n",edge[i].x,edge[i].y);
	}
 } 

你可能感兴趣的:(topsort)