食物链(AcWing,NOI 2001)

题目描述

题目链接:

https://www.acwing.com/problem/content/242/

思路：

这道题颇有难度，是经典的带权并查集问题，这里就不解释这些带权并查集的理论是怎么来的，就仅谈谈我对带权并查集的一些个人思考。仅表达大致的思想，可能有些地方写的不是那么严谨。

题目中会描述两种关系: x和y是同类关系，x吃y的关系。我们可以用“有向边”来统一处理这两种情况。

由题目中所说的A吃B，B吃C，C吃A，可知，我们这里定义的距离应当mod 3 

我们构造一个有向图G，图中若存在有向边x->y, 边权为a,则认为x和y连通，y和x也连通，y到x的边权为(3-a).

任意两个点x和y，若根据题目中的一些条件，能确定x和y之间的合法关系，则x和y应当属于同一个连通块(i.e. x和y之间会有一条有向通路)

为了后续统一维护信息，我们在每个连通块中取一个代表元素(并查集的根节点)，对该连通块中任意一个节点a的操作都转换为a所在连通块的代表元素b的操作。

fa[i] 维护i的父亲节点是谁

d[i] 维护i到fa[i]的有向距离是多少

下面来分析条件：

若1  x y,则令x向y 连一条边权为0的有向边

若2  x y,则令x向y连一条边权为1的有向边

考虑两种情况:

1. 在得到该条件之前，已经能够确定x和y的合法关系(i.e. x和y在同一个连通块中)，那么此时我们应当判断该条件是否是假话。

具体来说，考虑x，y所在的连通块中的代表元素u，考虑x到u的距离，y到u的距离，x到y的距离是否合法即可。

2. 在得到该条件之前，还不能够确定x和y的合法关系(i.e. x和y不在同一个连通块中)，那么此时我们应当将连接x所在的连通块和y所在的连通块并维护距离信息

具体来说，假设x所在连通块的代表元素为u，y所在连通块的代表元素为v

x到y的距离为dis，x到u的距离为d[x], y到v的距离为d[y]，那么可以推得: u到v 的距离应当是:(3-d[x]+dis+d[y])%3,再令fa[u]=v.

然后这道题还有个小技巧，就是在并查集的查询父亲节点的函数中，需要路径压缩，并更新d[N]数组的信息。

int get(int x){
    if(x==fa[x]) return x;
    int u=get(fa[x]);
    d[x]=(d[x]+d[fa[x]])%3;
    fa[x]=fa[fa[x]];
    return fa[x];
}

  在上面这个代码中，我们规定get(int x) 函数的返回值为x节点所在连通块的代表元素u(i.e. 并查集的根)，并且get函数具有副作用: 使得x的fa[x]=u，d[x]=x到u的有向距离。我们来看是如何实现这个事情的:

首先判断x本身是否为连通块的代表元素，这个是平凡的。

做一个递归，get(fa[x]), 根据我们对get函数的定义，此时fa[x]还是x的父亲节点，并且由于get函数的副作用，已经使得fa[fa[x]]=u, 且d[fa[x]]等于fa[x]到u的有向距离

那么我们只需要在此基础上维护一下x的信息就好了。

整个题目的代码：

#include<iostream>
#include<cstdio>
using namespace std;
const int N=1e5+10;
int fa[N];
int d[N];
int n,m;

void init(){
    for(int i=1;i<=n;i++){
        fa[i]=i;
        d[i]=0;
    }
    return ;
}
int get(int x){
    if(x==fa[x]) return x;
    int u=get(fa[x]);
    d[x]=(d[x]+d[fa[x]])%3;
    fa[x]=fa[fa[x]];
    return fa[x];
}

int main(void){

    scanf("%d%d",&n,&m);
    init();
    int ans=0;
    for(int i=1;i<=m;i++){
        int g,a,b;
        scanf("%d%d%d",&g,&a,&b);
        if(a>n || b > n ) {
            ans++;
            continue;
        }
        int dis=0;
        if(g==2) dis=1;
        int u=get(a);
        int v=get(b);
        if(u==v){
            int tmp=(dis+d[b])%3;
            if(tmp!=d[a]){
                ans++;
                continue;
            }
        }
        else{
            fa[u]=v;
            d[u]=(3-d[a]+dis+d[b])%3;
        }
    }
    printf("%d",ans);

    return 0;
}

在AcWing和洛谷上测试后均可以通过。