【BZOJ1179】【APIO2009】Atm（tarjan+spfa）

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

HINT

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

题解：
这道还好。就是把环缩点，跑最长路即可。因为有多个终点，所以有两种处理方法：逐个终点比较，或设一个虚拟终点T2。

代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<queue>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define N 500005
using namespace std;
int n,m,s,p; 
int e[N],nex[N],hd[N],e2[N],nex2[N],hd2[N],tot;
int dfn[N],low[N],tim,q[N],top,scc;
bool inq[N];
int bl[N],v[N],c[N];
int f[N];
void ins(int u,int v){e[++tot]=v,nex[tot]=hd[u],hd[u]=tot;}
void ins2(int u,int v){e2[++tot]=v,nex2[tot]=hd2[u],hd2[u]=tot;}
void tarjan(int x)
{
    int now=0;
    dfn[x]=low[x]=++tim;
    q[++top]=x,inq[x]=1;
    for(int i=hd[x];i;i=nex[i])
    {
        if(!dfn[e[i]]) 
        {
            tarjan(e[i]);
            low[x]=min(low[x],low[e[i]]);
        }
        else if(inq[e[i]]) low[x]=min(low[x],dfn[e[i]]);
    }
    if(low[x]==dfn[x])
    {
        scc++;
        while(now!=x)
        {
            now=q[top];top--;bl[now]=scc;
            v[scc]+=c[now];
            inq[now]=0;
        }
    }
}
void reb()
{
    tot=0;
    for(int i=1;i<=n;i++)
    for(int j=hd[i];j;j=nex[j])
    if(bl[i]!=bl[e[j]]) ins2(bl[i],bl[e[j]]);
}
void spfa()
{
    queue<int> Q;
    memset(inq,0,sizeof(inq));
    Q.push(bl[s]);inq[bl[s]]=1;f[bl[s]]=v[bl[s]];
    while(!Q.empty())
    {
        int now=Q.front();Q.pop();
        for(int i=hd2[now];i;i=nex2[i])
        {
            if(f[e2[i]]<f[now]+v[e2[i]])
            {
                f[e2[i]]=f[now]+v[e2[i]];
                if(!inq[e2[i]])
                {
                    inq[e2[i]]=1;
                    Q.push(e2[i]);
                }
            }
        }
        inq[now]=0;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1,u,v;i<=m;i++) scanf("%d%d",&u,&v),ins(u,v);
    for(int i=1;i<=n;i++) scanf("%d",&c[i]);
    for(int i=1;i<=n;i++) 
    if(!dfn[i]) tarjan(i);
    scanf("%d%d",&s,&p);
    reb();
    spfa();
    int ans=0;
    for(int i=1,x;i<=p;i++)
    {
        scanf("%d",&x);
        if(f[bl[x]]>ans) ans=f[bl[x]];  
    }
    printf("%d\n",ans);
    return 0;
}