Luogu P3627 抢掠计划

题目传送门

\(Tarjan\)缩点+SPFA最长路

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct zzz{
    int f,t,nex;
}e[500010]; int head[500010],tot;
void add(int x,int y){
    e[++tot].f=x;
    e[tot].t=y;
    e[tot].nex=head[x];
    head[x]=tot;
}
int dfn[500010],low[500010],deth,s[500010],top,col,belong[500010];
bool vis[500010];
void tarjan(int f){
    dfn[f]=low[f]=++deth;
    s[++top]=f; vis[f]=1;
    for(int i=head[f];i;i=e[i].nex){
        int to=e[i].t;
        if(!dfn[to]){
            tarjan(to); low[f]=min(low[f],low[to]);
        }
        else
          if(vis[to]) low[f]=min(low[f],dfn[to]);
    }
    if(dfn[f]==low[f]){
        col++;
        int k=0;
        do{
            k=s[top--];
            vis[k]=0;
            belong[k]=col;
        }while(k!=f);
    }
}
int read(){
    int k=0; char c=getchar();
    for(;c<'0'||c>'9';) c=getchar();
    for(;c>='0'&&c<='9';c=getchar())
      k=(k<<3)+(k<<1)+c-48;
    return k;
}
int atm[500010]; bool cof[500010];
int q[500010<<2],dis[500010]; int h=1,tail;
int main(){
    int n=read(),m=read();
    for(int i=1;i<=m;i++){
        int x=read(),y=read();
        add(x,y);
    }
    for(int i=1;i<=n;i++)
      if(!dfn[i]) tarjan(i);
    for(int i=1;i<=n;i++)
      atm[belong[i]]+=read();
    int s=read(),cnt=read();
    for(int i=1;i<=cnt;i++)
      cof[belong[read()]]=1;
    memset(head,0,sizeof(head)); tot=0;
    for(int i=1;i<=m;i++){
        int x=belong[e[i].f],y=belong[e[i].t];
        if(x!=y) add(x,y);
    }
    memset(vis,0,sizeof(vis));
    q[++tail]=belong[s]; dis[belong[s]]=atm[belong[s]];
    vis[belong[s]]=1;
    while(h<=tail){
        int k=q[h]; h++; vis[k]=0;
        for(int i=head[k];i;i=e[i].nex){
            int to=e[i].t;
            if(dis[k]+atm[to]>dis[to]){
                dis[to]=atm[to]+dis[k];
                if(!vis[to])
                  q[++tail]=to, vis[to]=1;
            }
        }
    }
    int ans=0;
    for(int i=1;i<=col;i++)
      if(cof[i]) ans=max(ans,dis[i]);
    cout<<ans;
    return 0;
}