poj 1988 并查集

并查集的简单应用，很多细节需要处理。处理好当前结点和祖先结点的距离关系。

//============================================================================
//
// >     File        :    poj1988.cpp
// >     Author      :    flowertree
// >     Time        :    2015年11月14日
// >     Algorithm   :    并查集路径压缩处理技巧 思路巧妙
//						  只需要处理栈底到祖先的关系即可 栈中其他节点作用是找祖先
//
//============================================================================

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
using namespace std;

#define MAX 300005

int f[MAX];
int num[MAX];
int d[MAX];

int find(int x)
{
	if(x == f[x])
		return x;
	else
	{
		int temp = find(f[x]);
		d[x] += d[f[x]];
		f[x] = temp;
		return f[x];
	}
}

void Union(int a, int b)
{
	int temp;
	int fa, fb;
	fa = find(a);
	fb = find(b);
	if(fa == fb)
		return ;
	else
	{
		d[fb] = num[fa];
		f[fb] = fa;
		num[fa] += num[fb];
	}
}

void Init()
{
	for(int i = 0; i < MAX; i++)
	{
		f[i] = i;
		num[i] = 1;
		d[i] = 0;
	}
}

int main()
{
	Init();
	int a, b;
	int m;
	char ch[5];
	scanf("%d", &m);
	while(m--)
	{
		scanf("%s", ch);
		if(ch[0] == 'M')
		{
			scanf("%d%d", &a, &b);
			Union(a, b);
		}
		else
		{
			scanf("%d", &a);
			int temp = find(a);
			printf("%d\n", num[temp] - d[a] - 1);
		}
	}
	system("pause");
	return 0;
}