uvalive 3027 Corporative Network(种类并查集)

112 阅读1分钟

题意:

题目就是要你输出各个点与根节点的距离。有两种指令

I a b,的意思:将b作为a的根节点,a和b之间的距离为|a-b|%1000。

E a, 的意思:求出a与根节点的距离。

思路:

并查集的变形,在压缩路径的时候顺便在更新一下与根节点的距离。

ps:记录与根节点的距离要再开一个数组,我在更新距离的时候还是写错了,太菜。

#include<iostream>
#include<cstdio>
#include<string>
#include<cmath>
using namespace std;
const int mx = 21000;
int fa[mx],d[mx];
int n;
void init(){
	for(int i = 1; i<mx ;i++){
		d[i] = 0;
		fa[i] = i;
	}
}
int find(int x){
	if(x != fa[x]){
		int te = fa[x];//一定要在压缩路径之前记录自己原先的父节点,方便更新
		fa[x] = find (fa[x]);
		d[x] += d[te];
	} 
	return fa[x];
}


 

int main(){
	int T;
	scanf("%d",&T); 
	while(T--){
		scanf("%d",&n);
		init();
		char s;
		int a, b;
		while(cin>>s && s != 'O'){
			if(s == 'E') {
				scanf("%d",&a);
				find(a);
			printf("%d\n",d[a]);
			}
			else {
				scanf("%d%d",&a,&b);
				fa[a] = b;
				d[a] = abs(a - b) % 1000;
			}
		}
	}	
}


\

本文已参与「新人创作礼」活动,一起开启掘金创作之路