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;
}