[Apio2009][bzoj1179]Atm

题意：一个n个点m条单向边的图，每个点有权值，给定出发点和p个可以停止的点，你可以随便走一条路径从出发点走到一个可以停止的点，但是每个点的点权只能计算一次，求能得到的最大权值。

n,m<=500000

题解：tarjan缩点，然后树形dp。显然强联通分量里每一个点都可以乱走（为所欲为）

#include<cstdio>
#include<iostream>
#define INF 2000000000
using namespace std;
inline int read()
{
int  x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();}
return x*f;
}

bool b[1000005],inq[500005];
int cnt=0,n,m,cc,s[1000005],dfn[500005],low[500005],rt,top=0,dn=0;
int head[1000005],f[1000005],q[500005],bel[500005];
struct edge{
int to,next;
}e[1000005];

void ins(int f,int t){e[++cnt].next=head[f];head[f]=cnt;e[cnt].to=t;}

void tarjan(int x)
{
dfn[x]=low[x]=++dn;q[++top]=x;inq[x]=1;
for(int i=head[x];i;i=e[i].next)
if(!dfn[e[i].to]){tarjan(e[i].to);low[x]=min(low[x],low[e[i].to]);}
else {if(inq[e[i].to]) low[x]=min(low[x],dfn[e[i].to]);}
if(dfn[x]==low[x])
for(++cc;q[top+1]!=x;bel[q[top]]=cc,s[cc]+=s[q[top]],b[cc]|=b[q[top]],inq[q[top--]]=0);
}

void dp(int x)
{
if(b[x])f[x]=s[x];else f[x]=-INF;
for(int i=head[x];i;i=e[i].next) {if(!f[e[i].to])dp(e[i].to);f[x]=max(f[x],f[e[i].to]+s[x]);}
}

int main()
{
cc=n=read();m=read();
for(int i=1;i<=m;i++){int u=read(),v=read();ins(u,v);}
for(int i=1;i<=n;i++)s[i]=read();
rt=read();m=read();
for(int i=1;i<=m;i++)b[read()]=1;
for(int i=1;i<=n;i++)if(!dfn[i])tarjan(i);
for(int i=1;i<=n;i++)for(int j=head[i];j;j=e[j].next)
if(bel[i]!=bel[e[j].to])ins(bel[i],bel[e[j].to]);
dp(bel[rt]);
cout<<f[bel[rt]];
return 0;
}