并查集 POJ1182 经典带权并查集

食物链

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 90562		Accepted: 27216

Description

动物王国中有三类动物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

 

食物链关系那我们就用r[i]数组来储存i与它的父亲节点p[i]的关系：0同类 1父亲吃孩子 2孩子吃父亲

那么find函数里的转换关系代码如下，注意，因为p[x]已经更新过了（从根节点一步步更新下来），所以应事先存下更新前的p[x]值，以便于关系的更新：r[i]=(r[i]+r[tem])%3;

后面的分类就是是否是一棵树上（同一集合里），是则判断，否则合并。

关系的运算有以下式子：

r[fx]=(r[y]-r[x]-(d-1)+3)%3;  （d为读入的运算

        fy

r[fx] / ` \

   fx        \

  /r[x]        \ r[y]

x  ----------  y

      (d-1)

画图推导出这样子的关系式我们即可进行判定即合并操作~

注：-1%3=-1！惊了！试了才知道！所以要把可能是负数的先变成非负数嗯+3再模

另外注意读题：是2情况下即表示x吃y时，x与y不能相同emmm嘤

 

//0同类 1父亲吃孩子 2孩子吃父亲
#include <cstdio>

const int maxn=50010;
const int maxk=100000;

int n;
long k;
int p[maxn],r[maxn];

int find(int x)
{
	if(x==p[x]){
		return x;
	}
	int tem=p[x];
	p[x]=find(p[x]);
	r[x]=(r[x]+r[tem])%3;
	return p[x];	//醉了一开始忘了写这一句了wa了好多发
}

int main()
{
	int d,x,y,fx,fy;
	long cnt=0;

	scanf("%d%ld",&n,&k);
	for(int i=1;i<=n;++i){
		p[i]=i;
		r[i]=0;
	}
	for(long i=0;i<k;++i){
		scanf("%d%d%d",&d,&x,&y);
		if(x>n||y>n||(d==2/*惊了!*/&&x==y)){
			++cnt;
			continue;
		}
		fx=find(x);fy=find(y);
		if(fx!=fy){
			p[fx]=fy;
			r[fx]=(r[y]-r[x]-(d-1)+3)%3;	//d-1是x与y的关系
		}
		else{
			if((r[y]-r[x]+3)%3!=(d-1)){		//有可能是负数!
				++cnt;
			}
		}
	}
	printf("%ld\n",cnt);

	return 0;
}