首页 > 代码库 > UVALive - 3027Corporative Network(带权并查集)

UVALive - 3027Corporative Network(带权并查集)

题目: UVALive - 3027Corporative Network(带权并查集)


题目大意:有n和节点,初始时每个节点的父节点都不存在,然后有下面两种操作:I 操作 I a,b 将a的父节点变成b。E操作 E a,查询a到它的父节点的距离。


解题思路:带权并查集。注意这里距离的变化是a -> b,那么a到根节点的距离就是a到b的距离的绝对值 % 1000 + b到它的根节点的距离。


代码:

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 2e5 + 5;
int p[maxn], c[maxn];
int n;

void init () {
	
	for (int i = 1; i <= n; i++) {
		p[i] = i;
		c[i] = 0;
	}
}

int getParent (int a) {
	
	if (a == p[a])
		return a;
	int t = p[a];
	p[a] = getParent (p[a]);
	c[a] += c[t];	
	return p[a];
}

int main () {

	int T;
	int a, b;
	char str[10];
	scanf ("%d", &T);
	while (T--) {

		scanf ("%d", &n);
		init();
		while (scanf ("%s", str) != EOF) {
			
			if (str[0] == 'O')
				break;
			if (str[0] == 'E') {

				scanf ("%d", &a);
				int q1 = getParent (a);
				printf ("%d\n", c[a]);
			} else {

				scanf ("%d%d", &a, &b);
				p[a] = b;
				c[a] = abs (a - b) % 1000;
			}
		}
	}
	return 0;
}

 

UVALive - 3027Corporative Network(带权并查集)