信息传递(带权并查集)
这一道题就是典型的带权并查集,首先每个点出度都是一,要回到本身的最小距离也就是环的最小长度,自然而然想到用并查集来维护。
再在更新之后才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); }
【推荐】还在用 ECharts 开发大屏?试试这款永久免费的开源 BI 工具!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步