POJ1988 - Cube Stacking - 并查集

并查集，带权，只需要find的时候多更新几个变量

#include<stdio.h>
struct node
{
	int p,dis,tot;
}num[30005];
int find(int n)
{
	int item;
	if(num[n].p==n)
	{
		return n;
	}
	item=find(num[n].p);
	num[n].dis+=num[num[n].p].dis;
	return num[n].p=item;
}
void move(int a,int b)
{
	int pa=find(a);
	int pb=find(b);
	if(pa!=pb)
	{
		num[pb].p=pa;
		num[pb].dis+=num[pa].tot;
		num[pa].tot+=num[pb].tot;
	}
}
int main()
{
	int n,a,b;
	while(scanf("%d",&n)!=EOF)
	{
		char ques;
		for(int i=0;i<30005;i++)
		{
			num[i].p=i;
			num[i].tot=1;
			num[i].dis=0;
		}
		while(n--)
		{
			getchar();
			scanf("%c",&ques);
			if(ques=='M')
			{
				scanf("%d%d",&a,&b);
				move(a,b);
			}
			else 
			{
				scanf("%d",&a);
				int pa=find(a);
				printf("%d\n",num[pa].tot-num[a].dis-1);
			}
		}
	}
}