信息传递(带权并查集)

这一道题就是典型的带权并查集,首先每个点出度都是一,要回到本身的最小距离也就是环的最小长度,自然而然想到用并查集来维护。

再在更新之后才get就剁手

 

复制代码
#include<bits/stdc++.h>
using namespace std;
int fa[200005],n,t,w[200005],mn=0x3f3f3f3f;
int get(int k){
    if(k==fa[k])return k;
    int last=fa[k];
    fa[k]=get(fa[k]);//先更新父节点才能获得父节点的w
    w[k]+=w[last];
    return fa[k];
}
void check(int x,int y){
    int fx=get(x),fy=get(y);
    if(fx==fy) mn=min(mn,w[x]+w[y]+1);//可以就更新
    else fa[fx]=fy,w[x]=w[y]+1;//如果不同就把它们连起来,到根节点的距离就是父节点再加一
    return;
}
int main(){
    //freopen("data.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=n;i++){
        scanf("%d",&t);
        check(i,t);//判断
    }
    printf("%d\n",mn);
}
复制代码

 

posted @   Coder_cjh  阅读(379)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示