[Apio2009]Atm 强联通分量+SPFA

原题链接
题意：有N个路口和M条单行道，告诉了每个路口ATM机中的钱数，市中心S的位置，P个酒吧的位置。求从市中心出发，途径ATM机获得钱（可重复走同一条路，同一个路口），最后在一个酒吧中消费，问最多可以获得多少钱。

思路：先找强联通分量进行缩点，构造新图。然后利用SPFA算法寻找从S出发的最长路径

AC代码：（取名是个坑，最好不要把变量名字取成库函数中的函数名，，，不然有可能CE）#include<iostream>
#include<cstdio>
#include<vector>
#include<stack>
#include<algorithm>
#include<cstring>
#include<queue>

#define maxn 500001

using namespace std;

vector<int> edge[maxn]; //原图
vector<int> edge1[maxn]; //缩点后的图
stack<int> s;
queue<int> q;

int dfn[maxn]; //dfn(u)为节点u搜索的次序编号(时间戳)
int low[maxn]; //low(u)为u或u的子树能够追溯到的最早的栈中节点的次序号。
int sccnum[maxn]; //记录每个点在第几号强连通分量里
int tclock;//记录时间戳
int num;//记录强联通分量个数
int mon[maxn]; //记录每个强联通分量钱数
int in[maxn],out[maxn]; //记录每个强连通分量的出入度
int money[maxn]; //ATM钱数
int dis[maxn]; //表示到i点最长路径
int vis[maxn]; //表示i是否在队列中

void tarjan(int u)
{
dfn[u]=low[u]=++tclock;
s.push(u);
for(int i=0;i<edge[u].size();i++)
{
int v=edge[u][i];
if(!dfn[v])
{
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(!sccnum[v]) //当前结点尚未构成强联通分量，即还在栈中
low[u]=min(low[u],dfn[v]);
}

if(low[u]==dfn[u]) //构成强连通分量
{
num++;
int v=-1;
while(v!=u)
{
v=s.top();
s.pop();
sccnum[v]=num;
mon[num]+=money[v];
}
}
}

void findscc(int n) //寻找强连通分量
{
tclock=num=0;
memset(dfn,0,sizeof(dfn));
memset(low,0,sizeof(low));
memset(sccnum,0,sizeof(sccnum));
memset(mon,0,sizeof(mon));
for(int i=1;i<=n;i++) //使每个点都遍历到
if(!dfn[i])
tarjan(i);
}

void spfa(int s)
{
memset(dis,-1,sizeof(dis));
memset(vis,0,sizeof(vis));
dis[s]=mon[s];
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
for(int i=0;i<edge1[u].size();i++)
{
if(dis[edge1[u][i]]<dis[u]+mon[edge1[u][i]])
{
dis[edge1[u][i]]=dis[u]+mon[edge1[u][i]];
if(!vis[edge1[u][i]])
{
vis[edge1[u][i]]=1;
q.push(edge1[u][i]);
}
}
}
}
}

int main()
{
int n,m;
while(~scanf("%d %d",&n,&m))
{
for(int i=1;i<=n;i++)
edge[i].clear();
int u,v;
while(m--)
{
scanf("%d %d",&u,&v);
edge[u].push_back(v);
}
for(int i=1;i<=n;i++)
scanf("%d",&money[i]);

findscc(n); //缩点

for(u=1;u<=n;u++) //构造新图
for(int i=0;i<edge[u].size();i++)
{
v=edge[u][i];
if(sccnum[u]!=sccnum[v])
edge1[sccnum[u]].push_back(sccnum[v]);
}

int S,P;
scanf("%d %d",&S,&P);

spfa(sccnum[S]); //找出到各个点的最长路径

int sum=0,k;
while(P--)
{
scanf("%d",&k);
sum=max(sum,dis[sccnum[k]]);
}

printf("%d\n",sum);
}
return 0;
}