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;
}