P3387 【模板】缩点

题目链接

题面：

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#define ll long long
using namespace std;
const int maxn=100100;
int head[maxn],ver[maxn],nt[maxn],val[maxn],tot1;
int dfn[maxn],low[maxn];
int st[maxn],ins[maxn],c[maxn];
int hh[maxn],vv[maxn],nn[maxn],va[maxn],tot2;
int n,m,num,top,cnt;
int dp[maxn],q[maxn],d[maxn];

void add1(int x,int y)
{
    ver[++tot1]=y,nt[tot1]=head[x],head[x]=tot1;
}

void add2(int x,int y)
{
    vv[++tot2]=y,nn[tot2]=hh[x],hh[x]=tot2,d[y]++;
}

void tarjan(int x)
{
    dfn[x]=low[x]=++num;
    st[++top]=x;
    ins[x]=1;
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        if(!dfn[y])
        {
            tarjan(y);
            low[x]=min(low[x],low[y]);
        }
        else if(ins[y])
            low[x]=min(low[x],dfn[y]);
    }

    if(dfn[x]==low[x])
    {
        cnt++;
        int z;
        do
        {
            z=st[top--];
            ins[z]=0;
            c[z]=cnt;
            va[cnt]+=val[z];
        }while(x!=z);
    }
}

int tp(void)
{
    int maxx=0;
    int l=1,r=0;
    for(int i=1;i<=n;i++)
    {
        dp[i]=va[i];
        maxx=max(maxx,dp[i]);
        if(d[i]==0) q[++r]=i;
    }

    while(l<=r)
    {
        int x=q[l++];
        for(int i=hh[x];i;i=nn[i])
        {
            int y=vv[i];
            if(--d[y]==0) q[++r]=y;
            dp[y]=max(dp[y],dp[x]+va[y]);
            maxx=max(maxx,dp[y]);
        }
    }
    return maxx;
}

int main(void)
{
    int x,y;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&val[i]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        add1(x,y);
    }
    for(int i=1;i<=n;i++)
        if(!dfn[i]) tarjan(i);
    for(int x=1;x<=n;x++)
    {
        for(int i=head[x];i;i=nt[i])
        {
            int y=ver[i];
            if(c[x]==c[y]) continue;
            add2(c[x],c[y]);
        }
    }
    printf("%d\n",tp());
    return 0;
}