这一道题就是典型的带权并查集,首先每个点出度都是一,要回到本身的最小距离也就是环的最小长度,自然而然想到用并查集来维护。
再在更新之后才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 ",mn); }