AcWing 240. 食物链

Posted qingyuyyyyy

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了AcWing 240. 食物链相关的知识,希望对你有一定的参考价值。

#include <iostream>
using namespace std;
const int N = 50010;
int n, m;
int p[N], d[N];  //p是baba,d是距离
int find(int x) {
    if (p[x] != x) {//如果x不是树根
        int t = find(p[x]);//指向根节点   存起来  因为赋值完之后就变了
        d[x] += d[p[x]];//更新为到根节点之间的距离 ==到父节点之间的距离+父节点到根节点之间的距离
        p[x] = t;//再把赋值成根节点
    }
    return p[x];    
}
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) p[i] = i;//初始化,自己是一个集合
    int res = 0;//假话的个数
    while (m -- ) {
        int t, x, y;//t表示询问的种类
        scanf("%d%d%d", &t, &x, &y);
        if (x > n || y > n) res ++ ;//超出范围,假话
        else {
            int px = find(x), py = find(y);//找根节点
            if (t == 1) {  //如果是同类
                if (px == py && (d[x] - d[y]) % 3)//如果在一个集合而且距离%3不等
                    res ++ ;//假话
                else if (px != py) {//如果根节点不同,说明不在一个集合
                    p[px] = py;//把py变成px的根节点
                    d[px] = d[y] - d[x];//因为同一类,所以%3之后相等,所以距离直接为差
                }
            } else {//表示吃   如果x吃y,那么d[x]-d[y]=1
                if (px == py && (d[x] - d[y] - 1) % 3)//如果在一个集合, 而且不为1
                    res ++ ;//假话
                else if (px != py) {//如果不在一个集合
                    p[px] = py;//合并
                    d[px] = d[y] + 1 - d[x];
                }
            }
        }
    }
    printf("%d
", res);
    return 0;
}

 

 

以上是关于AcWing 240. 食物链的主要内容,如果未能解决你的问题,请参考以下文章

# [acwing 240] 食物链 [并查集]

AcWing 240. 食物链

ACWing 240. 食物链 带权并查集

AcWing:240. 食物链(扩展域并查集)

240. 食物链(并查集+数论)

acwing3132. 食物(BZOJ3028)