poj1703(带权并查集)

最新推荐文章于 2020-05-10 22:03:47 发布

原创最新推荐文章于 2020-05-10 22:03:47 发布 · 630 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#poj

带权并查集专栏收录该内容

6 篇文章

订阅专栏

该博客主要介绍了如何使用带权并查集解决关于两个人属于两个敌对帮派关系的判断问题。通过题目描述和思路解析，博主展示了当输入两个人之间的敌对关系时，如何确定他们是否属于同一帮派、不同帮派或关系不确定，并在查找和合并操作中维护一个表示关系的额外数组。

题意：

有ｎ个人，属于两个敌对的帮派；

Ｏ　ａ　ｂ代表ａｂ是敌对的；

Ｓ　ａ　ｂ输出ａ，ｂ的关系（一个帮派，不同帮派或不确定）；

思路：

ｒ［ａ］　代表ａ和它父亲的关系；０为同帮，１为不同帮；

如果两个不再一个集合里，则就是不确定关系；

如果两个在同一个集合，就要判断他们和父亲的关系（同一个父亲）

如果两个人都和父亲是敌对．或者都是友好，则两人是友好；

否则敌对；需要在ｆｉｎｄ和Ｕｎｉｏｎ的时候维护ｒ数组；

#include<cstdio>
#include<cstring>

const int N = 50005;
int p[N],r[N],n,k;

void init(int x) {
	for(int i = 0; i <= x; i++) {
		p[i] = i;
		r[i] = 0;
	}
}

int find_set(int x) {
	if(x != p[x]) {
		int fx = find_set(p[x]);
		r[x] = (r[x] + r[p[x]]) % 3;
		p[x] = fx;
	}
	return p[x];
}
bool Union(int x,int y,int rela) {
	int px = find_set(x);
	int py = find_set(y);
	if(px == py) {
		if(((r[y] - r[x] + 3) % 3) != rela)
			return true;
		else
			return false;
	}
	p[py] = px;
	r[py] = (r[x] - r[y] + rela + 3) % 3;
	return false;

}
int main() {
	int sum = 0;
	scanf("%d%d",&n,&k);
	init(n);
	for(int i = 0; i < k; i++) {
		int tmp,x,y;
		scanf("%d%d%d",&tmp,&x,&y);
		if(x > n|| y > n || (x == y && tmp == 2))
			sum++;
		else if(Union(x,y,tmp - 1))
			sum++;
	}
	printf("%d\n",sum);
}