POJ 1182 食物链 并查集
http://poj.org/problem?id=1182
题目大意:
N只动物编号为1~N,所有动物都属于A,B,C中的一种,已知A吃B,B吃C,C吃A。
现在给你两种信息 1 x y表示x和y为同一种,2 x y表示x吃y。
给你K条这样的信息,让你判断有多少信息有错(x和y超过N也算错)
思路:
看书上的思路的。
可用一个并查集来维护。
对于每只动物i建立i-A,i-B,i-C,
1,.i-x表示i属于种类x。
2.并查集里的每一个组表示组内所有元素所有代表的情况可能同时发生或者不发生。
例如,如果i-A和j-B在同一个组里,就表示如果i属于种类A那么j一定属于种类B,如果j属于B那么i一定属于种类A,因此,对于每条信息,可做如下处理:
1.判断是否矛盾(看代码)
2. (1)x和y同一类 合并x-A和y-A,x-B和y-B,x-C和y-C
(2)x吃y 合并x-A和y-B,x-B和y-C,x-C和y-A。
#include<cstdio> const int MAXN=50000*3+10; int fa[MAXN]; int find(int cur) { return cur==fa[cur]?cur:fa[cur]=find(fa[cur]); } bool same(int x,int y) { return find(x)==find(y); } int main() { int n,k; scanf("%d%d",&n,&k); int v=n*3; for(int i=0;i<=v;i++) fa[i]=i; int ans=0; for(int i=0;i<k;i++) { int kind,x,y; scanf("%d%d%d",&kind,&x,&y); if(x>n||y>n) ans++; else if(kind==1) { if(same(x,y+n)||same(x,y+n+n)) ans++; else { int root_x=find(x),root_y=find(y); fa[root_x]=root_y; root_x=find(x+n),root_y=find(y+n); fa[root_x]=root_y; root_x=find(x+n+n),root_y=find(y+n+n); fa[root_x]=root_y; } } else { if(same(x,y)||same(x,y+n+n)) ans++; else { int root_x=find(x),root_y=find(y+n); fa[root_x]=root_y; root_x=find(x+n),root_y=find(y+n+n); fa[root_x]=root_y; root_x=find(x+n+n),root_y=find(y); fa[root_x]=root_y; } } } printf("%d\n",ans); return 0; }
新 blog : www.hrwhisper.me