不相交集 POJ1182 食物链

seagullyoyo

于 2014-08-15 08:53:17 发布

阅读量402

点赞数

CC 4.0 BY-SA版权

分类专栏：大力出奇迹

本文链接：https://blog.youkuaiyun.com/seagullyoyo/article/details/38579961

大力出奇迹专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决特定问题的方法。通过定义结构体存储节点信息，包括节点的父亲节点和关系属性，实现了节点集合的创建、查找及合并操作。在主函数中，通过对输入数据的处理，完成了对并查集的初始化、查找和合并操作，并解决了错误检测的问题。

#include <iostream>
using namespace std;

struct node
{
	int father;
	int relation;
};

node p[50010];

void make_set(int val)
{
	p[val].father = val;
	p[val].relation = 0;
}

void init()
{
	for(int i = 0; i < 50010; ++i)
		make_set(i);
}

int find_set(int val)
{
	if(val == p[val].father)
		return val;
	int pre = p[val].father;
	p[val].father = find_set(pre);
	p[val].relation = (p[val].relation + p[pre].relation) % 3;
	return p[val].father;		
}

int main(void)
{
	//freopen("1.txt", "r", stdin);
	int n, m, r, x, y;
	int rootx, rooty;
	int error = 0;
	init();
	cin >> n >> m;
	for(int k = 1; k <= m; ++k)
	{
		//cin >> r >> x >> y;	    // cin就TLE，scanf AC 
		scanf("%d%d%d", &r, &x, &y);
		if(x > n || y > n || (r == 2 && x == y))
		{
			error++;	
			continue;
		}
			
		rootx = find_set(x); rooty = find_set(y);
		
		if(rootx != rooty)
		{
			p[rootx].father = rooty;
			p[rootx].relation = (3 - p[x].relation + r-1 + p[y].relation) % 3;
		}
		else
		{
			if(r == 1 && p[x].relation != p[y].relation)
				error++;
			else if(r == 2 && (p[y].relation + 3-p[x].relation) % 3 != (4-r)%3)
				error++;
		}
	}
	cout << error << endl;

	return 0;
}