tarjan缩点

tarjan 缩点的方法：用tarjan求强连通分量，然后将同一个强连通分量中的点合成到同一个新节点中（需要借助一个belong数组）。

题目链接：P3387缩点

思路：先缩点，新节点的值等于强连通分量中所有节点权值之和。缩点之后就由原图得到了一张新图（其实是一棵树），然后再dfs，求出从根节点到叶子节点权值之和的最大值（dp）。

代码：

#include <cstdio>
#include <iostream>
#include <vector>
#include<cstring>
#define fori(a,b,c) for(int a=b;a<=c;a++)
#define ford(a,b,c) for(int a=c;a>=b;a--)
#define mem0(x) memset(x,0,sizeof(x))
#define mem1(x) memset(x,-1,sizeof(x))
using namespace std;
const int M=1e4+20;

int dfn[M],low[M],in[M],q[M],tot,top;
int n,m,a[M],b[M],belong[M],sum,vis[M];

//belong数组用来保存原节点合成到哪一个新节点中

vector<int> g1[M],g2[M];

void tarjan(int x)
{
    dfn[x]=low[x]=++tot;
    q[top++]=x;
    in[x]=1;
    for(int i=0;i<g1[x].size();i++)
    {
        int v=g1[x][i];
        if(!dfn[v])
        {
            tarjan(v);
            low[x]=min(low[x],low[v]);
        }
        else if(in[v])low[x]=min(low[x],low[v]);
    }
    if(low[x]==dfn[x])//一个强连通分量
    {
        sum++;
        int v;
        do{      
            v=q[--top];
            in[v]=0;
            belong[v]=sum;  //将同一个强连通分量中的节点合成到同一个新节点中
            b[sum]+=a[v];  
        }while(v!=x);
    }
}

void build()
{
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<g1[i].size();j++)
        {
            int v=g1[i][j];
            if(belong[i]!=belong[v])
            {
                g2[belong[i]].push_back(belong[v]);
            }
        }
    }
}

int dfs(int s)
{
    vis[s]=1;
    if(g2[s].size()==0)
        return b[s];
    int ans=-1;
    for(int i=0;i<g2[s].size();i++)
    {
        ans=max(ans,dfs(g2[s][i])+b[s]);
    }
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",a+i);
    int x,y;
    while(m--)
    {
        scanf("%d%d",&x,&y);
        g1[x].push_back(y);
    }
    for(int i=1;i<=n;i++)
        if(!dfn[i])tarjan(i);
    build();
    int ans=-1;
    for(int i=1;i<=sum;i++)
        if(!vis[i])ans=max(ans,dfs(i));
    printf("%d\n",ans);
    return 0;
}