poj 食物链

这里写图片描述

AC情况:这里写图片描述

思路:种类并查集。
这个题目只需要输入一组,可以写的简单一点,但是第一次做这种题目,我想加深点印象所以把函数都完整的写了一下,以后回顾也方便。
因为一共有3种动物,且是循环的吃的关系,一般情况下不能确定谁是第一种谁是第二种第三种,所以把每个动物i都分成3个元素:i-A、i-B、i-C。如果输入说法1 x y,则说明x-A=y-A、x-B=y-B、x-C=y-C。如果输入说法2 x y,则说明x-A=y-B、x-B=y-C、x-C=y-A。定义i代表i-A,i+n代表i-B,i+2n代表i-C。

#include<iostream>
using namespace std;
const int MAX_N = 50001;

int par[3*MAX_N],ran[3*MAX_N];

int i;
//初始化函数
void init(int n) {
    for (i = 0; i < n; i++) {
        par[i] = i;
        ran[i] = 0;
    }
}
//找元素x的父节点
int find(int x) {
    int k, j, r;
    r = x;
    while (r != par[r])r = par[r];
    k = x;
    while (k != r) {
        j = par[k];
        par[k] = r;
        k = j;
    }
    return r;
}
//合并元素x和y
void unite(int x, int y) {
    x = find(x);
    y = find(y);
    if (x == y)return;
    if (ran[x] < ran[y]) {
        par[x] = y;
    }
    else {
        par[y] = x;
        if (ran[x] = ran[y]) ran[x]++;
    }
}
//检验x和y的父节点是否相同
bool same(int x, int y) {
    return find(x) == find(y);
}

int main() {
    int n, k;
    cin >> n >> k;
    int t, x, y;
    init(n * 3);
    int ans = 0;
    for (i = 0; i < k; i++) {
        cin >> t >> x >> y;
        x--;           //把输入元素的范围调整为0~n-1,因为要根据par[x]寻找父节点,par是从0存的
        y--;
        if (x >=n || y >= n) {
            ans++;
            continue;
        }
        if (t == 1) {
            if (same(x, y + n) || same(x, y + 2 * n)) {
                ans++;

            }
            else {
                unite(x, y);
                unite(x + n, y + n);
                unite(x + 2*n, y +2* n);
            }
        }
        else {
            if (same(x, y) || same(x, y + 2 * n)) {
                ans++;
            }
            else {
                unite(x, y+n);
                unite(x + n, y + 2*n);
                unite(x + 2 * n, y );
            }
        }
    }
    cout << ans<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值