bzoj 1179: [Apio2009]Atm（Trajan+SPFA）

1179: [Apio2009]Atm

Time Limit: 15 Sec   Memory Limit: 162 MB
Submit: 4039   Solved: 1754
[ Submit][ Status][ Discuss]

Description

Input

第一行包含两个整数N、M。N表示路口的个数，M表示道路条数。接下来M行，每行两个整数，这两个整数都在1到N之间，第i+1行的两个整数表示第i条道路的起点和终点的路口编号。接下来N行，每行一个整数，按顺序表示每个路口处的ATM机中的钱数。接下来一行包含两个整数S、P，S表示市中心的编号，也就是出发的路口。P表示酒吧数目。接下来的一行中有P个整数，表示P个有酒吧的路口的编号

Output

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

Sample Input

6 7

1 2

2 3

3 5

2 4

4 1

2 6

6 5

10 12 8 16 1 5

1 4

4 3 5 6

Sample Output

47

思路还是很简单的

我们假设是有向无环图，那么很显然可以dp

求出起点到所有酒吧的最长路就是答案

但这道题有环，就要先将所有的强联通分量缩个点了

还是挺麻烦的

#include<stdio.h>
#include<vector>
#include<stack>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
vector<int> G[500005], Gp[500005];
stack<int> st;
queue<int> q;
int cnt, T, jb[500005], val[500005], fs[500005], low[500005], t[500005], vis[500005], scc[500005], dp[500005];
void Trajan(int u)
{
	int i, v;
	vis[u] = 1;
	st.push(u);
	t[u] = low[u] = ++T;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i];
		if(vis[v]==0)
		{
			Trajan(v);
			low[u] = min(low[u], low[v]);
		}
		else if(scc[v]==0)
			low[u] = min(low[u], t[v]);
	}
	if(low[u]==t[u])
	{
		cnt++;
		while(st.empty()==0)
		{
			v = st.top();
			st.pop();
			scc[v] = cnt;
			fs[cnt] += val[v];
			if(v==u)
				break;
		}
	}
}
int main(void)
{
	int n, m, i, j, x, y, S, p, ans;
	scanf("%d%d", &n, &m);
	for(i=1;i<=m;i++)
	{
		scanf("%d%d", &x, &y);
		G[x].push_back(y);
	}
	for(i=1;i<=n;i++)
		scanf("%d", &val[i]);
	scanf("%d%d", &S, &p);
	for(i=1;i<=p;i++)
	{
		scanf("%d", &x);
		jb[x] = 1;
	}
	Trajan(S);
	for(i=1;i<=n;i++)
	{
		for(j=0;j<G[i].size();j++)
		{
			x = G[i][j];
			if(scc[i]!=scc[x])
				Gp[scc[i]].push_back(scc[x]);
		}
	}
	memset(vis, 0, sizeof(dp));
	q.push(scc[S]);
	dp[scc[S]] = fs[scc[S]];
	vis[scc[S]] = 1;
	while(q.empty()==0)
	{
		x = q.front();
		q.pop();
		vis[x] = 0;
		for(i=0;i<Gp[x].size();i++)
		{
			y = Gp[x][i];
			if(dp[x]+fs[y]>dp[y])
			{
				dp[y] = dp[x]+fs[y];
				if(vis[y]==0)
				{
					q.push(y);
					vis[y] = 1;
				}
			}
		}
	}
	ans = 0;
	for(i=1;i<=n;i++)
	{
		if(jb[i])
			ans = max(ans, dp[scc[i]]);
	}
	printf("%d\n", ans);
	return 0;
}