【bzoj1179】[Apio2009]抢掠计划atm 强连通分量缩点＋spfa

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
HINT

50%的输入保证N, M<=3000。所有的输入保证N, M<=500000。每个ATM机中可取的钱数为一个非负整数且不超过4000。输入数据保证你可以从市中心沿着Siruseri的单向的道路到达其中的至少一个酒吧。

题目链接：
http://www.lydsy.com/JudgeOnline/problem.php?id=1179

思路：
可以经过点无数次，所以强连通分量中的点都可到达，权值为正，tarjan缩点后可以按点权spfa，枚举终点；

代码：

#include<iostream>
#include<stdio.h>
#include<stack>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
vector<int> lin[500005];
vector<int> d[500005];
int be[500005],dfn[500005],low[500005];
int a[500005];
int vis[500005];
int val[500005];
int dis[500005];
stack<int> qq;
int st;
int tim;
int ans;
queue<int> q;
int n,m;
int aa,bb;
int tot;
void spfa()
{
    for(int i=1;i<=tot;i++) vis[i]=0;   
    q.push(be[st]);
    dis[be[st]]=val[be[st]];
    vis[be[st]]=1;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        vis[now]=0;
        int len=d[now].size();
        for(int i=0;i<len;i++)
        {
            int v=d[now][i];
            dis[v]=max(dis[v],dis[now]+val[v]);
            if(!vis[v])
            {
                q.push(v);
                vis[v]=1;
            }
        }
    }
}
void tarjan(int x)
{

    dfn[x]=low[x]=++tim;
    qq.push(x);
    vis[x]=1;
    int len=lin[x].size();
    for(int i=0;i<len;i++)
    {
        int v=lin[x][i];
        if(!dfn[v])
        {
            tarjan(v);
            low[x]=min(low[x],low[v]);
        }  
        else if(vis[v]) low[x]=min(low[x],dfn[v]);
    //if(x==5) cout<<v<<" "<<vis[1]<<endl;
    }
    if(low[x]==dfn[x])
    {

        int now=0;
        tot++;
        while(now!=x)
        {
             now=qq.top();
            qq.pop();
            be[now]=tot;
            val[tot]+=a[now];
            vis[now]=0; 
        }
    }
//cout<<x<<" "<<dfn[x]<<low[x]<<endl;
}
void init()
{
    for(int i=1;i<=n;i++)
    {
        int len=lin[i].size();
        for(int j=0;j<len;j++)
        {
            if(be[i]!=be[lin[i][j]])
            {
                d[be[i]].push_back(be[lin[i][j]]);
            }
        }
    }

}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&aa,&bb);
        lin[aa].push_back(bb);
    }
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);

    for(int i=1;i<=n;i++)
    if(!dfn[i])
    tarjan(i);

    init();
    int p;
    scanf("%d%d",&st,&p);   
    //cout<<tot<<endl;
    spfa();
    for(int i=1;i<=p;i++)
    {
        scanf("%d",&aa);
        ans=max(ans,dis[be[aa]]);
    }
    printf("%d\n",ans);

}