POJ1082食物链

加权并查集入门习题。

传送门http://poj.org/problem?id=1182

下面来记录一下做法：

并查集的作用是询问两个对象时候在同一集合以及将两个非空不相交集合合并。

本题涉及两点之间是否能够两边，这样就可以生成多个集合。

而d=1 ，d=2则相当于将两个集合检索，若合法则合并，否则则将ans++。

但在检索时，我们发现采用一般的并查集无法实现（开三个除外），因为无法简单的利用两点是否在同一集合而判断是否可行。

故想到将其附加信息，而加权并查集的信息一般处理为儿子对父节点的关系或父节点对儿子的关系。

本题设rec[i]为i节点对fa[i]的关系。

如果通过a->b ,b->c能求出a->c，那么本题就变得十分简单。

通过枚举可以发现

令0表示同类，1表示被父节点吃，2表示吃父节点。

这样在求a->c 时，就可以用(rec[i]+rec[fa[i]])%3来表示。

下面上代码：

 1 #include "iostream"
 2 #include "cstdio"
 3 #include "cstring"
 4 #include "cctype"
 5 using namespace std;
 6 const int maxn = 50005;
 7 int n ,k ,fa[maxn] ,rec[maxn];
 8 void read(int &x)
 9 {
10     x = 0;
11     char c = getchar();
12     while(!isdigit(c))
13         c = getchar();
14     while(isdigit(c))
15     {
16         x = (x<<3) + (x<<1) + c - '0';
17         c = getchar();
18     }
19     return ;
20 }
21 
22 int find_(int x)
23 {
24     if(x == fa[x]) return x;
25     int temp = fa[x];
26     fa[x] = find_(fa[x]);
27     rec[x] = (rec[temp]+rec[x])%3;
28     return fa[x]; 
29 }
30 
31 int Union(int x ,int y ,int d)
32 {
33     int f1 = find_(x) ,f2 = find_(y);
34     if(f1 == f2)
35     {
36         if(d == 1)
37         {
38             if(rec[x] == rec[y])
39             return 0;
40             else return 1;
41         }
42         if(d == 2 && rec[x] != (rec[y] + 2)%3)
43           return 1;
44           return 0;
45     }
46     else
47     {
48         fa[f2] = f1;
49         rec[f2] = (rec[x] + (d-1) + (3-rec[y])) % 3;
50     }
51     return 0;                  
52 }
53 
54 int main()
55 {
56     read(n); read(k);
57     int num1 ,num2 ,num3 ,ans = 0;
58     for(int i=1; i<=n; i++) fa[i] = i ,rec[i] = 0;
59     for(int i=1; i<=k; i++)
60     {
61         read(num1);
62         read(num2);
63         read(num3);
64         if(num2>n || num3>n)
65         {
66             ++ans;
67             continue;
68         }
69         if(num1 == 2 && num2 == num3)
70         {
71             ++ans;
72             continue;
73         }
74         ans += Union(num2 ,num3 ,num1); 
75     }
76     printf("%d" ,ans);
77 }

 

posted @ 2017-11-23 10:39  Mnirvana  阅读(135)  评论(0)  编辑  收藏  举报