POJ 1182-食物链 并查集

题目来源: http://acm.pku.edu.cn/JudgeOnline/problem?id=1182

 

解题报告:

 

又是一道并查集的题, 但是比之前两道要稍微复杂点了.  做了很长时间, 搜索了其他的解题报告, 发现了一个很巧妙的方法

 

对每个结点i记录一个kind[i]代表结点i 对 其父节点p[i]之间的关系

 

kind[i]=1 父吃子(p[i]吃i) kind[i]=2 子吃父(i 吃 p[i])  kind[i]=0 同类 (i 和 p[i]同类)

 

于是对并查集的三种基本操作 makeSet, findSet, unionSet

 

应该有:

makeSet: 初始化每个结点的p[i]=i , kind[i]=0 rank[i]=0

 

findSet: 照例进行路径压缩, 但是压缩的同时, 结点的父节点也相应发生了变化, 因此 结点与其父节点之间的关系也发生了变化, 即kind[i]也应变化, 但是具体的关系可以根据 结点与其原父节点的关系以及其原父结点与其原父结点的父节点的关系推导出来, 即

                                 kind[a]=(kind[a]+kind[p[a]])%3

 

unionSet: 两个集合的合并,  其中一个集合的根节点指向另一个集合的根节点, 因此其中一个集合的根节点的kind也需相应的改变, 具体可以根据各个结点之间的关系推导出来, 不过, 我是通过枚举, 然后找规律得到的....

 

每次输入 d x y

对遇到的结点 x y , 先判断他们是否属于同一个集合, 如果属于, 那么他们的关系是确定的, 于是直接判断给出的d是否满足它们之间的关系即可

如果遇到的结点 x y, 它们不在一个集合, 那么它们的关系是不能确定的, 因此这句话是真的, 并且我们需要把它们的关系记录下来, 即调用unionSet来对两个结点所属的集合合并, 这样, 得到的新集合中, 所以的结点间的关系是确定的.

 

#include <iostream> using namespace std; int *p; int *rank; int *kind; //与其父亲结点之间的关系 //kind[p[i]]-kind[i]=0 同类; =1 父吃子; =2 子吃父 void makeSet(int a) { p[a]=a; rank[a]=0; kind[a]=0; } int findSet(int a) //路径压缩的同时,也要修改kind { int temp; if(a!=p[a]) { temp=findSet(p[a]); kind[a]=(kind[p[a]]+kind[a])%3; p[a]=temp; } return p[a]; } void exchange(int &x, int &y) { int temp=x; x=y; y=temp; } void _union(int a,int b,int d) { int roota=findSet(a); int rootb=findSet(b); if(rank[rootb]<rank[roota]) { exchange(roota,rootb); exchange(a,b); d=2-d; } p[roota]=rootb; //rootb成为新集合的代表, kind[roota]需要修改 if(rank[rootb]==rank[roota]) rank[rootb]++; switch(d) { case 0: //b吃a kind[roota]=(kind[b]-kind[a]+1)%3; break; case 1: //a,b同类 kind[roota]=(kind[b]-kind[a]+3)%3; break; case 2: //a吃b kind[roota]=(kind[b]-kind[a]+2)%3; break; } } int main() { int N,K; int n=0; cin >> N >> K; p=new int[N+1]; rank=new int[N+1]; kind=new int[N+1]; for(int i=1;i<=N;i++) makeSet(i); int d,x,y; while(K--) { scanf("%d%d%d",&d,&x,&y); if(x>N || y>N) n++; else { if(d==2 && x==y) n++; else { if(findSet(x)==findSet(y)) { if((kind[y]-kind[x]+1-d)%3!=0) n++; } else _union(x,y,d); } } } cout << n << endl; return 0; }

 

附录:

 

食物链

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 18371		Accepted: 5208

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