POJ 1182 带权并查集

传送门：http://poj.org/problem?id=1182

要点：1.理解并查集这种数据结构：

整体而言，并查集的实现过程是将图中    开始位于独立集合的节点      中有联系的点归类到同一集合的过程。

（1）查操作（find_root）：找目标节点的根节点，递归或循环实现。并且每次将目标节点的父节点改为根节点。即做路径压缩的操作（本质限制树的深度，从而优化时间复杂度）。

（2）并操作（union_set）：找到两节点的根节点，然后，将一个根的根节点改为另一个根。

时间复杂度：建立一个集合的时间复杂度为O(1)，N次合并M查找的时间复杂度为O(M Alpha(N))，这里Alpha是Ackerman函数的某个反函数，在很大的范围内（人类目前观测到的宇宙范围估算有10的80次方个原子，这小于前面所说的范围）这个函数的值可以看成是不大于4的，所以并查集的操作可以看作是线性的。

2.理解题目：

题目中一共有三种关系，同类、吃、被吃可看做三种不同的权重0、1、2.。各种关系由向量表示（方向为从父节点指向子节点），权重表示子节点与父节点的关系，1、2分别表示父吃子、子吃父。（路径压缩的过程中别忘记改对应的权重）

首先做查操作：

（1）如果两节点位于同一集合，则判断根、x、y三节点间向量和模3的值为零即可。（可自己在纸上画一画）

（2）如果两节点不再同一集合，做并操作。

 

代码附上：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 50500;
int parent[maxn];
int val[maxn];

void init(int n)                        //初始化父节点、权重数组
{
	for(int i = 0; i < n; i++)
	{
	    parent[i] = i;
	    val[i] = 0;
	}
}

int find_root(int i)                    //查操作
{
	if(i == parent[i]) return i;
	int y = find_root(parent[i]);
 	val[i] = (val[i] + val[parent[i]]) % 3; //改权重，因为下一步要进行路径压缩了
    parent[i] = y;                       //路径压缩
	return y;
}

void union_set(int d, int x, int y)     //并操作
{
	int root_x = find_root(x);
	int root_y = find_root(y);
	parent[root_y] = root_x;
	val[root_y] = ( (3 - val[y]) + (d - 1) + val[x] ) % 3;      //四个点间向量的关系，手动先在纸上模拟
}

bool judge(int d, int x, int y, int n)
{
	if(x > n || y > n) return false;
	int root_x = find_root(x);
	int root_y = find_root(y);
	if(root_x == root_y)
	{
		if( (val[x] + (d - 1) +  (3 - val[y]) ) % 3 == 0) return true;     //三个点向量和为0
		else return false;
	}
	else
	{
		union_set(d, x, y);
		return true;
	}
}
int main()
{

	int n, k;
	int d, x, y;
	int num = 0;
	scanf("%d%d", &n, &k);
	init(n);
	for (int i = 0 ; i < k ; i ++)
	{
		scanf("%d %d %d", &d, &x, &y);
		if(!judge(d, x, y, n)) num ++;
	}
	printf("%d\n",num);
}