【ybtoj 3.4强连通分量课堂过关】A.有向图缩点 【tarjan】【DP】【拓扑】

题目链接：有向图缩点

分析

先做tarjan缩点，然后枚举每一条边，看这条边上的点在不在同一个强连通分量上，然后一边拓扑一边DP。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;

int n,m,ans,tm,tt;
int a[10001],dfn[10001],col[10001],low[10001];
int st[10001],top,ru[10001];
int f[10001],sum[10001];
int hd[10001],tot,hd1[10001],tot1;

struct lwx
{
	int to,next;
}e[100001],e1[100001];

void add(int x,int y)
{
	tot++;
	e[tot]=(lwx){y,hd[x]};
	hd[x]=tot;
}

void add1(int x,int y)
{
	tot1++;
	e1[tot1]=(lwx){y,hd1[x]};
	hd1[x]=tot1;
}

void tarjan(int x)
{
	dfn[x]=low[x]=++tm;
	st[++top]=x;
	for(int i=hd[x];i>0;i=e[i].next)
    {
    	int v=e[i].to;
    	if(!dfn[v])
    	{
    		tarjan(v);
    		low[x]=min(low[x],low[v]);
		}
		else if(!col[v])
		{
			low[x]=min(low[x],low[v]);
		}
	}
	if(dfn[x]==low[x])
	{
		col[x]=++tt;
		sum[tt]+=a[x];
		while(st[top]!=x)
		{
			sum[tt]+=a[st[top]];
			col[st[top--]]=tt;
		}
		top--;
	}
}

void dp()
{
	queue<int> q;
	for(int i=1;i<=tt;i++)
	{
		if(!ru[i])
		{
			q.push(i);
			f[i]=sum[i];
		 } 
	}
	while(!q.empty())
	{
		int y=q.front();
		q.pop();
		for(int i=hd1[y];i>0;i=e1[i].next)
		{
			int v=e1[i].to;
			f[v]=max(f[v],f[y]+sum[v]);
			ru[v]--;
			if(!ru[v]) q.push(v);
		}
	}
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	for(int i=1;i<=m;i++)
	{
		int x,y;
		cin>>x>>y;
		add(x,y);
	}
	for(int i=1;i<=n;i++)
	{
		if(!dfn[i]) tarjan(i);
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=hd[i];j>0;j=e[j].next)
		{
			int v=e[j].to;
			if(col[i]!=col[v])
			{
				add1(col[i],col[v]);
				ru[col[v]]++;
			}
		}
	}
	dp();
	for(int i=1;i<=n;i++)
	{
		ans=max(ans,f[i]);
	}
	cout<<ans;
	return 0;
}