缩点+spfa最长路【bzoj】 1179: [Apio2009]Atm

【bzoj】 1179: [Apio2009]Atm

Description

Siruseri 城中的道路都是单向的。不同的道路由路口连接。按照法律的规定， 在每个路口都设立了一个 Siruseri 银行的 ATM 取款机。令人奇怪的是，Siruseri 的酒吧也都设在路口，虽然并不是每个路口都设有酒吧。Banditji 计划实施 Siruseri 有史以来最惊天动地的 ATM 抢劫。他将从市中心 出发，沿着单向道路行驶，抢劫所有他

途径的 ATM 机，最终他将在一个酒吧庆 祝他的胜利。使用高超的黑客技术，他获知了每个 ATM 机中可以掠取的现金数额。他希 望你帮助他计算从市中心出发最后到达某个酒吧时最多能抢劫的现金总数。他可 以经过同一路口或道路任意多次。但只要他抢劫过某个 ATM 机后，该 ATM 机 里面就不会再有钱了。

Input

第一行包含两个整数N、M。N表示路口的个数，M表示道路条数。

接下来M行，每行两个整数，这两个整数都在1到N之间，

第i+1行的两个整数表示第i条道路的起点和终点的路口编号。

接下来N行，每行一个整数，按顺序表示每个路口处的ATM机中的钱数。

接下来一行包含两个整数S、P，S表示市中心的编号，也就是出发的路口。P表示酒吧数目。

接下来的一行中有P个整数，表示P个有酒吧的路口的编号

N, M<=500000。每个ATM机中可取的钱数为一个非负整数且不超过4000。

输入数据保证你可以从市中心沿着Siruseri的单向的道路到达其中的至少一个酒吧。

Output

输出一个整数，表示Banditji从市中心开始到某个酒吧结束所能抢劫的最多的现金总数。

莫名其妙的自信自己可以一遍A，毕竟做过几乎一样的加强版（捂脸。。。）

先缩点，把强连通分量里的每一个点的权值全部赋给所之后的点，也就是belong[i]。

再对于缩完点之后的图，跑一个点权SPFA就可以了。

拓扑排序+DP的做法会更快，不过有待学习。

code：

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int wx=500017;
inline int read(){
	int sum=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0';ch=getchar();}
	return sum*f;
}
struct e{
	int nxt,to,dis;
}edge[wx];
struct E{
	int nxt,to;
}Edge[wx];
int n,m;
int num,Tot,top,tot,col;
int a[wx],A[wx],belong[wx],size[wx],dis[wx],vis[wx];
int head[wx],Head[wx],dfn[wx],low[wx],st[wx],flag[wx];
void add(int from,int to,int dis){
	edge[++num].nxt=head[from];
	edge[num].to=to;
	edge[num].dis=dis;
	head[from]=num;
}
void add(int from,int to){
	Edge[++Tot].nxt=Head[from];
	Edge[Tot].to=to;
	Head[from]=Tot;
}
void Tarjan(int u){
	dfn[u]=low[u]=++tot;
	st[++top]=u;
	for(int i=Head[u];i;i=Edge[i].nxt){
		int v=Edge[i].to;
		if(!dfn[v]){
			Tarjan(v);
			low[u]=min(low[u],low[v]);
		}
		else if(!belong[v]){
			low[u]=min(low[u],dfn[v]);
		}
	}
	if(dfn[u]==low[u]){
		belong[u]=++col;
		size[col]++;
		A[col]+=a[u];
		while(st[top]!=u){
			belong[st[top]]=col;
			size[col]++;
			A[col]+=a[st[top]];
			top--;
		}
		top--;
	}
}
void CQ(){
	for(int u=1;u<=n;u++){
		for(int i=Head[u];i;i=Edge[i].nxt){
			int v=Edge[i].to;
			if(belong[u]!=belong[v]){
				add(belong[u],belong[v],0);
			}
		}
	}
}
queue<int > q;
void SPFA(int s){
	for(int i=1;i<=col;i++)dis[i]=-1;
	dis[s]=A[s];vis[s]=1;q.push(s);
	while(q.size()){
		int u=q.front();q.pop();
		vis[u]=0;
		for(int i=head[u];i;i=edge[i].nxt){
			int v=edge[i].to;
			if(dis[v]<dis[u]+A[v]){
				dis[v]=dis[u]+A[v];
				if(!vis[v]){
					vis[v]=1;q.push(v);
				}
			}
		}
	}
}
int main(){
	n=read();m=read();
	for(int i=1,x,y;i<=m;i++){
		x=read();y=read();add(x,y);
	}
	for(int i=1;i<=n;i++)a[i]=read();
	int s,p;s=read();p=read();
	for(int i=1,x;i<=p;i++){
		x=read();flag[i]=x;
	}
	for(int i=1;i<=n;i++){
		if(!dfn[i])Tarjan(i);
	}
	CQ();
	SPFA(belong[s]);
	int ans=0;
	for(int i=1;i<=p;i++){
		ans=max(ans,dis[belong[flag[i]]]);
	}
	printf("%d\n",ans);
	return 0;
}