带权并查集实战：2001年NOI食物链问题详解

版块：SEO软文类型：普通作者：用意要查看：9 回复：0 获赞：0 时间：2025-07-01 11:13:12

截图未命名.jpg 带权并查集实战：2001年NOI食物链问题详解并查集 NOI竞赛食物链问题第1张

一、问题背景与算法思路

食物链问题需要维护三种生物关系（同类、捕食、被捕食），判断K条陈述中有多少是假话。这个问题完美展现了带权并查集在关系维护中的强大能力。

二、完整代码实现（带详细注释）

#include <iostream>
#include <vector>
using namespACe std;

const int MAXN = 50010;

class UnionFind {
private:
    vector<int> parent;   // 父节点数组
    vector<int> rank;     // 秩数组
    vector<int> relation; // 关系数组：0-同类 1-吃父节点 2-被父节点吃
    
public:
    // 构造函数：初始化并查集
    UnionFind(int n) : parent(n+1), rank(n+1, 0), relation(n+1, 0) {
        for(int i = 0; i <= n; ++i) {
            parent[i] = i; // 初始时每个元素的父节点是自己
        }
    }
    
    // 查找根节点（带路径压缩和关系维护）
    int find(int x) {
        if(parent[x] != x) {
            int orig_parent = parent[x];
            parent[x] = find(parent[x]); // 路径压缩
            relation[x] = (relation[x] + relation[orig_parent]) % 3; // 更新关系
        }
        return parent[x];
    }
    
    // 合并集合（带关系维护）
    bool unite(int x, int y, int type) {
        int rootX = find(x);
        int rootY = find(y);
        
        if(rootX == rootY) { // 已在同一集合
            // 检查关系是否矛盾
            if((relation[x] - relation[y] + 3) % 3 != type)
                return false;
            return true;
        }
        
        // 按秩合并（小树合并到大树）
        if(rank[rootX] > rank[rootY]) {
            parent[rootY] = rootX;
            relation[rootY] = (relation[x] - relation[y] - type + 6) % 3;
        } else {
            parent[rootX] = rootY;
            relation[rootX] = (relation[y] - relation[x] + type + 3) % 3;
            if(rank[rootX] == rank[rootY]) {
                rank[rootY]++; // 秩相同时合并后秩+1
            }
        }
        return true;
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int N, K;
    cin >> N >> K;
    
    UnionFind uf(N);
    int falseCount = 0;
    
    for(int i = 0; i < K; ++i) {
        int type, x, y;
        cin >> type >> x >> y;
        
        // 输入合法性检查
        if(x > N || y > N || (type == 2 && x == y)) {
            falseCount++;
            continue;
        }
        
        // 关系类型转换：1->0(同类), 2->1(捕食)
        int rel = type - 1;
        
        if(!uf.unite(x, y, rel)) {
            falseCount++;
        }
    }
    
    cout << falseCount << endl;
    return 0;
}