poj 1182 食物链 带权并查集经典模板(NOI2001)

版权声明:点个关注(^-^)V https://blog.csdn.net/weixin_41793113/article/details/88936708

Description

动物王国中有三类动物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),输出假话的总数。 

Input

第一行是两个整数N和K,以一个空格分隔。 
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。 
若D=1,则表示X和Y是同类。 
若D=2,则表示X吃Y。

Output

只有一个整数,表示假话的数目。

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3

Source

Noi 01

扫描二维码关注公众号,回复: 5787287 查看本文章

1.如果在同一棵树中find(x) == find(y):直接判断是否说谎。

    1)如果 d ==1,那么 x 与 y 应该是同类,他们的r[]应该相等

         如果不相等,则说谎数 +1

    2)如果 d==2,那么 x 应该吃了 y,也就是 (r[x]+1)%3 == r[y]

         如果不满足,则说谎数 +1

         如何判断 x 吃了 y 是  (r[x]+1)%3 == r[y],请看下图:(PS:箭头方向指向被吃方)

这种循环的关系,可以联想到求模

这里写图片描述

思路:把确定了相对关系的节点放在同一棵树中

            每个节点对应的 r[]值记录他与根节点的关系:

             0:同类, 

             1:被父亲节点吃, 

             2: 吃父亲节点

        (a, b) 0:同类 、  1:a被b吃 、 2:a吃b

  (x, y)

(y,  z)

 (x,z)

如何判断

0

0

0

0+0 = 0

0

1

1

0+1 = 1

0

2

2

0+2 = 2

1

0

1

1+0 = 1

1

1

2

1+1 = 2

1

2

0

(1+2)%3 = 0

2

0

2

2+0 = 2

2

1

0

(2+1)%3 = 0

2

2

1

(2+2)%3 = 1

关于合并时r[]值的更新

如果 d == 1则 x和y 是同类 ,那么 y 对 x 的关系是 0

如果 d == 2 则 x 吃了 y,  那么 y 对 x 的关系是 1, x 对 y 的关系是 2.

综上所述 ,无论 d为1 或者是为 2,  y 对 x 的关系都是 d-1

定义 :fx 为 x 的根点, fy 为 y 的根节点

           合并时,如果把 y 树合并到 x 树中

           如何求 fy 对 fx 的r[]关系?

           fy 对 y 的关系为 3-r[y]

           y  对 x 的关系为 d-1

           x  对 fx 的关系为 r[x]

           所以 fy 对 fx 的关系是(3-r[y] + d-1 + r[x])%3

 不带权的并查集find+union 路径压缩模板

int find(int x){
    if(f[x]==x)
        return x;
    return f[x] = find(f[x]);
}
void Union(int x,int y){//小写的union和c++的内置冲突
    int a = find(x);
    int b = find(y);
    if(a!=b)
        f[a] = b;
}

带权的并查集find+union 路径压缩模板 r[x]权值 0是同类,1被父亲吃,2吃父亲


int find(int x){//找父亲节点
    if(x == p[x])
        return x;
    int t = p[x];
    p[x] = find(p[x]);
    r[x] = (r[x]+r[t])%3; //回溯由子节点与父节点的关系和父节点与根节点的关系找子节点与根节点的关系
    return p[x];
}

void Union(int x, int y, int d){
    int a = find(x);
    int b = find(y);
 
    p[b] = a; //合并树 注意:被 x 吃,所以以 x 的根为父
    r[b] = (r[x]-r[y]+3+(d-1))%3; //对应更新与父节点的关系
}

对比了上面带权和不带权并查集之后,你会发现其实带权的并查集看起来也不是那么懵逼= =

#include<cstdio>
 
const int maxn = 50000+10;
 
int p[maxn]; //存父节点
int r[maxn];//存与父节点的关系 0 同一类,1被父节点吃,2吃父节点
 
void init(int n){ //初始化
    for(int x = 1; x <= n; x++){
        p[x] = x; //开始自己是自己的父亲节点
        r[x] = 0;//开始自己就是自己的父亲,每一个点均独立
    }
}
 
int find(int x){ //找父亲节点
    if(x == p[x]) return x;
 
    int t = p[x];
    p[x] = find(p[x]);
    r[x] = (r[x]+r[t])%3; //回溯由子节点与父节点的关系和父节点与根节点的关系找子节点与根节点的关系
    return p[x];
}
 
void Union(int x, int y, int d){
    int fx = find(x);
    int fy = find(y);
 
    p[fy] = fx; //合并树 注意:被 x 吃,所以以 x 的根为父
    r[fy] = (r[x]-r[y]+3+(d-1))%3; //对应更新与父节点的关系
}
 
int main(){
    int n, m;
    scanf("%d%d", &n, &m);
    init(n);
 
    int ans = 0;
    int d, x, y;
    while(m--){
        scanf("%d%d%d", &d, &x, &y);
 
        if(x > n || y > n || (d == 2 && x == y)) ans++; //如果节点编号大于最大编号,或者自己吃自己,说谎
 
        else if(find(x) == find(y)){//如果原来有关系,也就是在同一棵树中,那么直接判断是否说谎
            if(d == 1 && r[x] != r[y]) ans++; //如果 x 和 y 不属于同一类
            if(d == 2 && (r[x]+1)%3 != r[y]) ans++; // 如果 x 没有吃 y (注意要对应Uinon(x, y)的情况,否则一路WA到死啊!!!)
        }
        else 
            Union(x, y, d); //如果开始没有关系,则建立关系
    }
    printf("%d\n", ans);
    return 0;
}

ACM博客_kuangbin POJ 1182 食物链(并查集)

猜你喜欢

转载自blog.csdn.net/weixin_41793113/article/details/88936708