0
点赞
收藏
分享

微信扫一扫

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

zidea 2023-03-29 阅读 18


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



举报

相关推荐

0 条评论