[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;
}