并查集 AcWing 240. 食物链

并查集 AcWing 240. 食物链

原题链接

AcWing 240. 食物链

算法标签

并查集

思路

用并查集维护每条食物链之间的关系， 对于每类动物之间， 采用该动物距离根节点之间的距离来表示动物之间的关系
具体的见下图

对于该并查集 需要额外维护距离

代码即

if(p[x]!=x){
    // u为祖宗节点
    int u=find(p[x]);
    // 更新 当前点距祖宗节点距离 = 当前点距父节点距离 + 父节点距祖宗节点距离
    d[x]+=d[p[x]];
    // 更新 当前点父亲节点 = 祖宗节点 
    p[x]=u;
}

对于同类关系
若不属于同一条食物链
需要进行并查集合并
合并时新边权值为

对于捕食关系
若不属于同一条食物链
需要进行并查集合并
合并时新边权值为
具体思路见代码

代码

#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 100005;
// p存储节点祖宗节点 d存储当前节点据祖宗节点的距离
int p[N], d[N];
inline int read(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
// 查询祖宗节点
int find(int x){
    if(p[x]!=x){
        // u为祖宗节点
        int u=find(p[x]);
        // 更新 当前点距祖宗节点距离 = 当前点距父节点距离 + 父节点距祖宗节点距离
        d[x]+=d[p[x]];
        // 更新 当前点父亲节点 = 祖宗节点 
        p[x]=u;
    }
    return p[x];
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n=read(), m=read();
    rep(i, 1, n+1){
        p[i]=i;
    }
    int res=0;
    while(m--){
        int t=read(), x=read(), y=read();
        if(x>n||y>n){
            res++;
        }else{
            int px=find(x), py=find(y); 
            if(t==1){
                // 如果在一条食物链且距离 % 3为1， 2 表明不是同类
                if(px==py&&(d[x]-d[y])%3){
                    res++;
                }else if(px!=py){/* 如果不在同一条食物链
                    进行食物链合并 
                    对于合并边的权值为 
                    为保证x, y属于同一类 
                    需要使(x到py的距离 + 合并边的权值 - y到py的距离) % 3 = 0   
                    即x到py的距离 = y到py的距离 */
                    p[px]=py;
                    d[px]=d[y]-d[x];
                }
            }else{
                // 如果在一条食物链且距离 - 1 % 3为1， 2 表明不是捕食关系
                if(px==py&&(d[x]-d[y]-1)%3){
                    res++;
                }else if(px!=py){/* 如果不在同一条食物链
                    进行食物链合并 
                    对于合并边的权值为 
                    为保证x, y属于为捕食关系
                    需要使(x到py的距离 + 合并边的权值 - y到py的距离 - 1) % 3 = 0*/
                    p[px]=py;
                    d[px]=d[y]+1-d[x];
                }
            }
        }
    }
    printf("%lld", res);
    return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈