pku1988并查集

 

题意：共有1-30000这些元素，每次M a b就是将a所在的栈放在b所在的栈的前面，C a表示查询a的下面有几个元素。。。10^6个查询

 

 

分析：开始没想清楚怎么存他们之间的个数关系，也想过反向建并查集（其实无论反向还是正向都要维护两个数组。。。）， 后来直接正向建并查集，记录根节点所在的集合的元素个数num[find(i)]，记录每个元素在这个集合当中排第几d[i],则对某个元素而言，其结果就是num[find(i)]-d[i]。。。。更新总是在find函数里面实现。。注意更新过程中一定要先更新父节点，后更新子节点。。。

 

 

代码：

#include<iostream>
#include<algorithm>
using namespace std;

const int N=30010;
int n, p, f[N], d[N], num[N];

int findset(int x)
{
	if(f[x]!=x)
	{
		int tmp = f[x];
		//d[x] += d[f[x]];
		f[x] = findset(f[x]);
		if(f[tmp]!=tmp)
			d[x] += d[tmp]-1;
	}
	return f[x];
}

void unionset(int x, int y)
{
	int fx = findset(x);
	int fy = findset(y);
	if(fx!=fy)
	{
		f[fy] = fx;
		d[fy] = num[fx]+1;
		num[fx] += num[fy];
	}
}

int main()
{
	int i, j, k, x, y;
	char op[3];
	while(scanf("%d", &p)!=EOF)
	{
		for(i=0; i<N; i++)
		{
			f[i] = i;
			d[i] = 1;
			num[i] = 1;
		}
		for(i=0; i<p; i++)
		{
			scanf("%s", op);
			if(op[0]=='C')
			{
				scanf("%d", &x);
				y = findset(x);
			//	printf("%d %d %d..\n", num[y], d[x], num[y]-d[x]);
				printf("%d\n", num[y]-d[x]);
			}
			else
			{
				scanf("%d%d", &x, &y);
				unionset(x, y);
			}
		}
	}
	return 0;
}