poj1182(带权并查集)

最新推荐文章于 2020-03-02 17:09:10 发布

二分查找

最新推荐文章于 2020-03-02 17:09:10 发布

阅读量388

点赞数

CC 4.0 BY-SA版权

分类专栏：带权并查集文章标签： poj

本文链接：https://blog.youkuaiyun.com/yeyeyeguoguo/article/details/45483815

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

6 篇文章

订阅专栏

本文介绍了一道关于带权并查集的经典算法题目，通过实现特定的关系更新逻辑来解决父子节点间的关系判断问题。文章提供了完整的C++代码示例，展示了如何初始化并查集、查找集合代表元素及进行集合合并的过程。

题意：

中文题就不描述了；

思路：

带权并查集模板题；

加入一个数组ｒ，表示这个点和它父节点的关系；

０表示同类，１表示父亲吃它，２表示它吃父亲；

每次需要更新和父亲之间的关系；

#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);
}