题目:洛谷3389
缩点本质上就是把一些可以相互到达(强连通分量)的点看成一个点,然后构建一个新图,方便计算。
#include<iostream>
using namespace std;
const int N=10005;
const int M=100005;
int n,m,head[N],dfn[N],low[N],vis[N],s[N],top,cnt,ans;
int Head[N],w[N],D[N],F[N],tot,in[N];//D[]:DP数组,F[]:记录那几个点属于同一个强连通分量,in[]:入度
struct node{
int x,y,next;
}e[M],ne[M];//旧图 / 新图
void Tarjan(int u){
dfn[u]=low[u]=++cnt,vis[u]=1,s[++top]=u;
for(int i=head[u],v;v=e[i].y,i;i=e[i].next)//Tarjan
if(!dfn[v]){
Tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(vis[v]) low[u]=min(low[u],dfn[v]);
if(low[u]==dfn[u])//合并
while(int k=s[top--]){
vis[k]=0,F[k]=u;//记录从属关系
if(k==u) break;
w[u]+=w[k];//点权合并
}
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>w[i];
for(int i=1,x,y;i<=m;i++){
cin>>x>>y;
e[i].x=x,e[i].y=y,e[i].next=head[x],head[x]=i;
}
for(int i=1;i<=n;i++)
if(!dfn[i]) Tarjan(i);
for(int i=1,x,y;x=F[e[i].x],y=F[e[i].y],i<=m;i++)
if(x!=y)//如果x,y不属于同一个强连通分量
ne[++tot].x=x,ne[tot].y=y,ne[tot].next=Head[x],Head[x]=tot,in[y]++;//建立新图
top=0;
for(int i=1;i<=n;i++)
if(F[i]==i&&!in[i]) s[++top]=i,D[i]=w[i];//入度为0即可作为起点
while(top){
int k=s[top--];
for(int i=Head[k],v;v=ne[i].y,i;i=ne[i].next){//DP
D[v]=max(D[v],D[k]+w[v]);
if(--in[v]==0) s[++top]=v;
}
}
for(int i=1;i<=n;i++) ans=max(ans,D[i]);
cout<<ans;
return 0;
}