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),输出假话的总数。
首先我们要明确并查集的作用:快速判断两个数是否同一集合与快速合并两个集合成一个集合并求出一些节点之间的关系,根据的就是树的特点:每个孩子节点有且仅有一个父节点。这样就用数组记录父节点就还(根就记录自己),合并操作就是合并两个根节点,这儿有个优化就是启发式合并:根记录孩子个数,并把个数少的并到个数大的上面。
不过我们有其他大招:路径压缩,即我们每次查询的时候都把一条线上的所有节点连接到祖先节点,这样每次查找都很快(并查集在路径压缩之后的时间复杂度是阿克曼函数)。依据就是我们只需要知道多个孩子节点的祖先是否一致就能判断是否一个集合,不需要知道树上的结构。我们的权值则是一般记录此节点与父节点的关系,只要满足这个关系可以传递我们就可以模仿矢量计算来处理权值。
这儿我们要明确是有三种关系的:两者同类,吃父节点,被父节点吃,所以权值可以用0,1,2表示
注意有个关键就是当我们知道x与祖先x1的关系,y与祖先y1的关系,x与y的关系时,求x1与y1的关系时,使用矢量 计算:
x1->x ->y ->y1 计算
/*n个动物 k句话 有一种循环a吃b 吃c c吃a 开始不知道n种动物关系是什么 两种询问:d=1 x y为同类 d=2 x吃y 判断假话条数(关键之违背之前的关系) 并查集可以很好解决的满足区间传递关系的区间合并问题,注意一般是多棵树*/ #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int Max=50010; int fat[Max],ran[Max]; void Init(int n)//初始化重要 { for(int i=0; i<=n; i++) { fat[i]=i;//初始化都是指向(看做)自己 ran[i]=0;//0同类 1吃父节点 2被父节点吃 } return; } int Find(int x)//找寻父节点+路径压缩 { if(x==fat[x]) return fat[x]; int y=Find(fat[x]); ran[x]=(ran[x]+ran[fat[x]])%3;//递归后从祖先节点向后到每个孩子来计算 return fat[x]=y;//路径压缩 } int Union(int typ,int x,int y)//区间并与查询 { int x1=Find(x); int y1=Find(y); if(x1==y1)//共父节点才能判断出关系 { if((ran[x]-ran[y]+3)%3==typ-1) return 0; return 1; } fat[x1]=y1;//连接两父节点 ran[x1]=(-ran[x]+typ-1+ran[y]+3)%3;//使用类似向量方法来计算权值,虽然题目只有两个,但是会出现被吃这种情况,所以要变成3种情况,注意一定要处理负数的情况 return 0; } int main() { int n,k,ans; int typ,smt1,smt2; scanf("%d %d",&n,&k); Init(n); ans=0; for(int i=0; i<k; i++) { scanf("%d %d %d",&typ,&smt1,&smt2); if(smt1==smt2&&typ==2) ans++; else if(smt1>n||smt2>n) ans++; else ans+=Union(typ,smt1,smt2); } printf("%d\n",ans); return 0; }