【模板】强连通分量

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/m0_38083668/article/details/83063043

题目：缩点

代码：

#include <bits/stdc++.h>
using namespace std;

const int Maxn=10005;
const int Maxm=100010;
int n,m,size,Index,cnt,tot,ans;
int low[Maxn],num[Maxn],fa[Maxn],vis[Maxn],q[Maxn];
int first[Maxn],First[Maxn],f[Maxn],a[Maxn],to[Maxn],sum[Maxn];
struct shu{int to,next;}edge[Maxm<<1],Edge[Maxm<<1];

inline int get_int()
{
    int x=0,f=1;
    char c;
    for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
    if(c=='-') f=-1,c=getchar();
    for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
    return x*f;
}

inline void build(int x,int y)
{
    edge[++size].next=first[x],first[x]=size,edge[size].to=y;
}
inline void Build(int x,int y)
{
    Edge[++size].next=First[x];First[x]=size,Edge[size].to=y;
}

inline void tarjan(int p)
{
    low[p]=num[p]=++Index,vis[p]=1,q[++tot]=p;
    for(int u=first[p];u;u=edge[u].next)
    {
      int to=edge[u].to;
      if(!num[to]) tarjan(to),low[p]=min(low[p],low[to]);
      else if(vis[to]) low[p]=min(low[p],num[to]);
    }
    if(low[p]==num[p])
    {
      cnt++;
      while(1)
      {
      	int x=q[tot--];
      	fa[x]=cnt,vis[x]=0,sum[cnt]+=a[x];
      	if(x==p) break;
      }
    }
}

inline void dfs(int p)
{
    if(~f[p]) return;
    f[p]=sum[p];
    for(int u=First[p];u;u=Edge[u].next)
    {
      int to=Edge[u].to;
      dfs(to);
      f[p]=max(f[p],f[to]+sum[p]);
    }
}

int main()
{
    memset(f,-1,sizeof(f));
    n=get_int(),m=get_int();
    for(int i=1;i<=n;i++) a[i]=get_int();
    for(int i=1;i<=m;i++)
    {
      int x=get_int(),y=get_int();
      build(x,y);
    }
    for(int i=1;i<=n;i++) if(!num[i]) tarjan(i);
    size=0;
    for(int i=1;i<=n;i++)
      for(int u=first[i];u;u=edge[u].next)
        if(fa[i]!=fa[edge[u].to]) Build(fa[i],fa[edge[u].to]),to[fa[edge[u].to]]++;
    for(int i=1;i<=cnt;i++) if(!to[i]) dfs(i);
    for(int i=1;i<=cnt;i++) ans=max(ans,f[i]);
    cout<<ans;
    return 0;
}