传送门: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);
}