【洛谷P2661】信息传递 (updated)

题目大意:给定一棵 N 个节点的内向树森林,求该内向树森林的最小环的大小(按边计算)。

题解:先删链,再计算环的大小,统计答案即可。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;

int n,ans,to[maxn],indeg[maxn];
bool vis[maxn];

void read_and_parse(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&to[i]),++indeg[to[i]];
}

void del(int u){
    vis[u]=1;
    if(!--indeg[to[u]])del(to[u]);
}

int cyc(int u,int dep){
    vis[u]=1;
    if(vis[to[u]])return dep;
    else return cyc(to[u],dep+1);
}

void solve(){
    ans=0x3f3f3f3f;
    for(int i=1;i<=n;i++)if(!indeg[i]&&!vis[i])del(i);
    for(int i=1;i<=n;i++)if(indeg[i]&&!vis[i])ans=min(ans,cyc(i,1));
    printf("%d\n",ans);
}

int main(){
    read_and_parse();
    solve();
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/wzj-xhjbk/p/10202268.html