(用树结构支持并查集8.2.2)POJ 1988 Cube Stacking(并查集的使用:判断一个集合中元素的个数---->>某一个元素的下面还有多少个元素)

/*
 * POJ_1988.cpp
 *
 *  Created on: 2013年11月5日
 *      Author: Administrator
 */

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int maxn =  100010;

int set[maxn];
int cnt[maxn];
int top[maxn];

int set_find(int p){
	if(set[p] < 0){
		return p;
	}

	if(set[set[p]] >= 0){//若set[p]下方还有元素,则调整p所在栈的最下面的元素
		int fa = set[p];
		set[p] = set_find(fa);
		cnt[p] += cnt[fa];//累加fa到set[p]之间的元素的个数
	}

	return set[p];
}


void set_join(int x,int y){
	x = set_find(x);
	y = set_find(y);
	set[x] = y;
	set_find(top[y]);//刷新y所在栈原先的栈顶元素导y之间的元素的个数
	cnt[x] = cnt[top[y]] + 1;
	top[y] = top[x];//更新y的所在栈的站定元素为x原先所在栈的栈顶元素
}


int main(){
	memset(set,-1,sizeof(set));
	memset(cnt,0,sizeof(cnt));
    int i;
    for(i = 0 ; i < maxn ; ++i){
    	top[i] = i;
    }

	int p;
	scanf("%d",&p);

	char str[10];
	while(p--){
		scanf("%s",str);

		if(str[0] == 'M'){
			int x,y;
			scanf("%d%d",&x,&y);
			set_join(x,y);
		}else{
			int x;
			scanf("%d",&x);
			set_find(x);
			printf("%d\n",cnt[x]);
		}
	}

	return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

帅气的东哥

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值