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),输出假话的总数。
如果father[A]=B ,rela[A]=0表示A与B同类;rela[A]=1 表示B可以吃A;rela[A]=2 表示A可以吃B。
然后就是枚举出所有情况,就可以推知公式,最好不要每种权值分开写,不然会乱。
第一道带权并查集,感觉不容易推公式,弄出来结果了,自然好用了。
#include <cstdio> #include <cstring> #include <vector> #include <cmath> #include <stack> #include <cstdlib> #include <queue> #include <map> #include <iostream> #include <algorithm> using namespace std; typedef long long LL; int n,k,ans; int d,x,y; int father[50100],rela[50100]; int findfa(int x)//找祖宗顺便明确这个点与祖宗的关系 { if (x!=father[x]) { int zu=findfa(father[x]);//找祖宗 rela[x]=(rela[x]+rela[father[x]])%3;//枚举后可以得出 father[x]=zu; } return father[x]; } int combine(int op,int x,int y) { int zux=findfa(x);//找祖宗顺便明确x与祖宗的关系 int zuy=findfa(y);//找祖宗顺便明确y与祖宗的关系 if (zux==zuy)//在一个体系中 { if ((rela[y]-rela[x]+3)%3!=op) return false;//枚举后可以得出 else return true; } else//是两个体系,就一定能通过关系的旋转对接上 { father[zuy]=zux;//让祖宗对上 rela[zuy]=(rela[x]-rela[y]+op+3)%3;//枚举后可以得出 return true; } } int main() { scanf ("%d%d",&n,&k); for (int i=1;i<=n;i++) father[i]=i; while (k--) { scanf ("%d%d%d",&d,&x,&y); if (x>n||y>n||(d==2&&x==y)) ans++; else if (!combine(d-1,x,y)) ans++; } printf ("%d\n",ans); return 0; }