【并查集】食物链

题目传送门:http://codevs.cn/problem/1074/

现在才发现我这么蒟蒻,并查集都不怎么会用啊!!!

这道题似乎用了一个很巧妙(估计是我太弱)的东西,就是用 i , i+n , i+n*2 来表示 i 在 A B C 三个种类,似乎是可以轮换的

然后每读入一次先验证是否是假话,不是的话就合并(好像是废话)

具体见代码吧

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

#define LL long long

int fa[150005],n,m,x,y,z,ans;

int find(int x){
    if (fa[x]==x) return x;
    fa[x]=find(fa[x]);
    return fa[x];
}

void unite(int x,int y){
    x=find(x);
    y=find(y);
    if (x==y) return;
    fa[x]=y;
}

int main(){
    cin>>n>>m;

    for (int i=1;i<=n*3;i++)
        fa[i]=i;
    for (int i=1;i<=m;i++){
        cin>>z>>x>>y;

        if (x<0 || x>n || y<0 ||y>n){
            ans++;
            continue;
        }

        if (z==1){
            if (find(x)==find(y+n) || find(x)==find(y+n*2)){//x与y一样时显然不能互相吃来吃去
                ans++;
                continue;
            }
            else{
                unite(x,y);
                unite(x+n,y+n);
                unite(x+n*2,y+n*2);
            }
        }

        if (z==2){
            if (find(x)==find(y) || find(x)==find(y+n*2)){//x吃y时x显然不与y一样,也不能是y吃x
                ans++;
                continue;
            }
            else{
                unite(x,y+n);
                unite(x+n,y+n*2);
                unite(x+n*2,y);
            }
        }
    }

    cout<<ans;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
并查集是一种用于解决集合合并与查询问题的数据结构。在食物链问题中,我们可以使用并查集来判断给定的K句话中有多少句是假话。 首先,我们需要创建一个并查集,其中每个动物都是一个节点。初始时,每个节点都是独立的集合。 然后,我们按照给定的K句话进行处理。对于每一句话,我们需要判断它是真话还是假话。 如果是第一种说法"1 X Y",表示X和Y是同类。我们可以通过将X和Y所在的集合合并来实现。即将X所在的集合的根节点指向Y所在的集合的根节点,或者将Y所在的集合的根节点指向X所在的集合的根节点。 如果是第二种说法"2 X Y",表示X吃Y。我们需要判断X和Y是否属于同一类。如果它们属于同一类,那么这句话就是假话。否则,这句话是真话。 最后,我们统计假话的总数即可。 以下是一个示例代码,演示了如何使用并查集解决食物链问题: ```python class UnionFind: def __init__(self, n): self.parent = list(range(n)) self.rank = [0] * n def find(self, x): if self.parent[x] != x: self.parent[x] = self.find(self.parent[x]) return self.parent[x] def union(self, x, y): root_x = self.find(x) root_y = self.find(y) if root_x == root_y: return if self.rank[root_x] < self.rank[root_y]: self.parent[root_x] = root_y elif self.rank[root_x] > self.rank[root_y]: self.parent[root_y] = root_x else: self.parent[root_y] = root_x self.rank[root_x] += 1 def count_false_statements(N, K, statements): uf = UnionFind(N+1) false_count = 0 for statement in statements: type, X, Y = statement if X > N or Y > N: false_count += 1 continue if type == 1: if uf.find(X) == uf.find(Y+N) or uf.find(X+N) == uf.find(Y): false_count += 1 else: uf.union(X, Y) uf.union(X+N, Y+N) uf.union(X+2*N, Y+2*N) elif type == 2: if uf.find(X) == uf.find(Y) or uf.find(X+N) == uf.find(Y): false_count += 1 else: uf.union(X, Y+N) uf.union(X+N, Y+2*N) uf.union(X+2*N, Y) return false_count N = 5 K = 7 statements = [(1, 1, 2), (2, 1, 3), (1, 2, 3), (2, 2, 4), (1, 4, 1), (1, 4, 3), (2, 4, 3)] false_count = count_false_statements(N, K, statements) print(false_count) # 输出:3 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值