E - 食物链 POJ - 1182(带权并查集)
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
Input
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
Output
只有一个整数,表示假话的数目。
Sample Input
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
Sample Output
3
推荐阅读:https://blog.csdn.net/niushuai666/article/details/6981689
补充两张图:
图一:当a,b同根的时候,判断a,b的关系是否正确。
图二:当不同根时,要合并两块集合,注意权值的维护
代码:
#include <iostream>
#include <stdio.h>
using namespace std;
int pre[50005];
int date[50005];
int d,x,y,n,k;
int a,b,c;
int ans = 0;
int root(int x)
{
if ( x!=pre[x] ) {
int t = pre[x];
pre[x] = root(t);
date[x] = (date[x]+date[t])%3; // 压缩路径时的维护权值
}
return pre[x];
}
void join( int a, int b, int d )
{
int x = root(a);
int y = root(b);
if ( x==y ) {
if ( (date[b]+3-date[a])%3!=d-1 ) { // 见图一
ans ++;
}
}
else {
pre[y] = x;
date[y] = ( d-1+date[a]+3-date[b] )%3; // 见图二
}
}
int main()
{
scanf("%d %d",&n,&k);
for ( int i=1; i<=n; i++ ) {
pre[i] = i;
date[i] = 0;
}
for ( int i=0; i<k; i++ ) {
scanf("%d %d %d",&a,&b,&c);
if ( b>n || c>n || (b==c&&a==2) ) {
ans ++;
continue ;
}
join(b,c,a);
}
printf("%d\n",ans);
return 0;
}