POJ 1182 食物链

 1 #include <iostream>
 2 #include <algorithm>
 3 #include <cmath>
 4 #include <stdio.h>
 5 #include <cstring>
 6 #include <string>
 7 #include <cstdlib>
 8 #include <queue>
 9 #include <stack>
10 #include <set>
11 #include <vector>
12 #include <map>
13 #include <list>
14 #include <iomanip>
15  #include <fstream>
16 
17 using namespace std;
18 int fa[50009],eat[50009];
19 //不要忘记初始化数组惹qwq 
20 void initailize(int n)
21 {
22     for(int i=0;i<n;++i)
23     {
24         fa[i]=i;//一开始自立门户 
25         eat[i]=0;//食性是莫得食性的 
26      } 
27 }
28 int get(int x)//查询根节点
29 {
30     int temp;
31     if(x==fa[x])
32         return x;//找到根节点 
33         
34     temp=fa[x];//没有找到根节点，但是找到了上一级 
35     fa[x]=get(temp);//更新自己的根节点，找到了上一级的根节点
36     //捕食关系更新，根据自己祖宗的食性知道自己的食性
37      eat[x]=(eat[x]+eat[temp])%3;//哇呜，传说中的矢量运算了？ 
38      //暂时规定其中的关系数字 0 1 2 分别代表同类/捕食/被捕食 
39      return temp;
40 }
41 
42 int main()
43 {
44     int n,k,cnt=0;
45     scanf("%d%d",&n,&k);
46     initailize(n);//不知道有没有拼对，反正是初始化的意思 
47     for(int i=1;i<=k;++i)
48     {
49         int d,x,y;
50         scanf("%d%d%d",&d,&x,&y);
51         //注意特殊情况（条件2、3），小动物的编号都已经超过了n
52          if(x>n||y>n||(x==y&&d==2))//我~吃~我~自~己~
53              {cnt++;continue;}
54         int gx=get(x),gy=get(y);//先找一下自己祖宗 
55         //如果他们祖宗不一样，那就合并成一个（所以一直合并的最终结果就是长度为2的树咯？） 
56         if(gx!=gy)
57         {
58             //emm毕竟一般来说都是x吃y，所以x是y的上级祖宗，加之它是有方向的矢量，所以方便编程就是
59             fa[y]=gx;
60             //d-1=实际捕食关系， 矢量相减就是食性距离，加3防止数字变负 
61             eat[y]=((d-1)+eat[x]-eat[y]+3)%3;
62             continue;
63         }
64         //他们已经是一个祖宗了 
65         else{
66                 if(d==1)//同类 
67                 {
68                     if(eat[x]!=eat[y])//同类的食性居然不一样？！ 
69                         {cnt++;continue;}
70                 }
71                 if(d==2)//x吃y，注意这是矢量加减（据称;滑稽），所以有方向 
72                 {
73                     if((eat[x]+1)%3!=eat[y])//x吃y但是之前说x与y是同类/被捕食的关系 
74                     {
75                         {cnt++;continue;}
76                     }
77                 }
78         }
79     }
80     printf("%d\n",cnt);
81     return 0;
82 }