首页 > 代码库 > poj 1182 食物链(种类并查集 ‘初心者’)
poj 1182 食物链(种类并查集 ‘初心者’)
题目链接:http://poj.org/problem?id=1182
借着这题可以好好理解一下种类并查集,这题比较简单但挺经典的。
题意就不解释了,中问题。
关于种类并查集结局方法也是挺多的
1扩增倍数。
就是有几种关系就建立几倍数组,具体方法在这里不详细说了,这种方法有弊端
比较复杂而且内存消耗比较大如果关系比较多就容易爆掉。
2向量的方法。
这种方法要详细解说一下。这个方法好处都有啥.......(自行脑补后面的话)
这个方法的优点占用内存比较小而且比较简洁。只要找到关系就行。
下面就用方法2来说一下这道题目
这题总共有3种关系
1)同类。2)A eat B。3)B eat A。
所以就设root[i],等于1表示同类,等于2表示关系2,等于3表示关系3
初始化是将root的值全定义为0表示他们毫不相关,然后再慢慢将关系加入进去
int find(int x) {
if(x == f[x])
return x;
int t = find(f[x]);
root[x] = (root[x] + root[f[x]]) % 3;//这个注意一下在寻找根节点的过程中要记得更新一下root的值。
f[x] = t;
return f[x];
}//寻找根节点
int a = find(x) , b = find(y);
if(d == 1) {
if(a == b) {
if(root[x] != root[y])//这个很好理解就不解释了
count++;
}
else {
f[a] = b;
root[a] = root[y] - root[x];//root[a]+root[x]=root[y] 这样就好理解了吧
root[a] = (root[a] + 3) % 3;
}
}
if(d == 2) {
if(a == b) {
if((root[x] + 1) % 3 != root[y])//这个也很好理解就是A->B or B->C or C->A他们的root关系就差1
count++;
}
else {
f[a] = b;
root[a] = root[y] - root[x] - 1;//root[a]+root[x] +1 = root[y];
root[a] = (root[a] + 3) % 3;
}
}
//这些都是关键代码
#include <iostream>#include <cstring>#include <cstdio>using namespace std;const int M = 5e4 + 10;int n , k , root[M] , f[M];void init() { for(int i = 1 ; i <= n ; i++) { f[i] = i , root[i] = 0; }}int find(int x) { if(x == f[x]) return x; int t = find(f[x]); root[x] = (root[x] + root[f[x]]) % 3; f[x] = t; return f[x];}int main() { int d , x , y; scanf("%d%d" , &n , &k); int count = 0; init(); while(k--) { scanf("%d%d%d" , &d , &x , &y); if(x > n || y > n) { count++; continue; } int a = find(x) , b = find(y); if(d == 1) { if(a == b) { if(root[x] != root[y]) count++; } else { f[a] = b; root[a] = root[y] - root[x]; root[a] = (root[a] + 3) % 3; } } if(d == 2) { if(a == b) { if((root[x] + 1) % 3 != root[y]) count++; } else { f[a] = b; root[a] = root[y] - root[x] - 1; root[a] = (root[a] + 3) % 3; } } } printf("%d\n" , count); return 0;}
poj 1182 食物链(种类并查集 ‘初心者’)