POJ1988 Cube Stacking

最新推荐文章于 2014-02-17 06:29:00 发布

qhpeklh5959

最新推荐文章于 2014-02-17 06:29:00 发布

阅读量343

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/severus_qin/article/details/8454064

版权

数据结构专栏收录该内容

41 篇文章 0 订阅

订阅专栏

题目链接：http://poj.org/problem?id=1988

这道题当指令是M的时候，就是并查集的合并工作了，C的时候自然就是查找，但是麻烦就是怎么去弄明白那个x下面到底有多少个元素，而且它是整列整列的弄的，所以需要找个东西标记一下。

用一个域u标记元素x的上面有多少个元素，用一个域d标记元素x的下面有多少个元素，合并的时候只要更新这两个域内的值就行了，那么对于每一个查找指令，答案就是x所在集合根节点下面的元素个数减去元素x上面的元素个数再减去x本身咯～不过这个真心不好想啊～

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 30001
struct data
{
	int p, u, d;
}p[N];
int find(int x)
{
	int t;
	if (p[x].p != x)
	{
		t = p[x].p;
		p[x].p = find(t);
		p[x].u += p[t].u;
	}
	return p[x].p;
}
int main()
{
	int P;
	int x, y;
	int r1, r2;
	char c;
	for (int i = 1; i < N; i++)
	{
		p[i].p = i;
		p[i].u = 0;
		p[i].d = 1;
	}
	cin >> P;
	while (P--)
	{
		scanf("\n%c", &c);
		if (c == 'M')
		{
			scanf(" %d %d", &x, &y);
			r1 = find(x); r2 = find(y);
			if (r1 != r2)
			{
				p[r2].p = r1;
				p[r2].u = p[r1].d;
				p[r1].d += p[r2].d;
			}
		}
		if (c == 'C')
		{
			scanf(" %d", &x);
			y = find(x);
			printf("%d\n", p[y].d - p[x].u - 1);
		}
	}
	return 0;
}