bzoj1179 [Apio2009]Atm[tarjan缩点]

Description

Siruseri城中的道路都是单向的。不同的道路由路口连接。按照法律的规定，在每个路口都设立了一个Siruseri银行的ATM取款机。令人奇怪的是，Siruseri的酒吧也都设立在路口，虽然并不是每个路口都设有酒吧。
Banditji计划实施Siruseri有史以来最惊天动地的ATM抢劫。他将从市中心出发，沿着单向道路行驶，抢劫所有他途经的ATM机，最终他将在一个酒吧庆祝他的胜利。
使用高超的黑客技术，他获知了每个ATM机中可以掠取的现金数额。他希望你帮助他计算从市中心出发最后到达某个酒吧时最多能抢劫的现金总数。他可以经过同一路口或道路任意多次。但只要他抢劫过某个ATM机后，该ATM机里面就不会再有钱了。
例如，假设该城中有6个路口，道路的连接情况如下图所示：
市中心在路口1，由一个入口符号→来标识，那些有酒吧的路口用双圈来表示。每个ATM机中可取的钱数标在了路口的上方。在这个例子中，Banditji能抢劫的现金总数为47，实施的抢劫路线是：1→2→4→1→2→3→5。

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的单向的道路到达其中的至少一个酒吧。

没看题解还一次AC，感觉真是爽爆了！！！(≥◇≤)
原题的Description是张图，我手打下来了…
一个强连通分量里的奥特曼肯定都能访问，考虑到这个性质，我们可以利用tarjan缩点。如果这个强连通分量里有一个奥特曼有酒吧，这个强连通分量就有酒吧。
缩完点后可以dfs跑一遍，也可以spfa一遍，因为缩完点后肯定是个无环图，访问了一个节点不可能再访问。如果访问到的节点有酒吧，就把ans与dis[now]比较，更新就行。
感觉缩点的题都好简单…缩完点就和图论模板题一样…
代码：

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int head1[500005],head2[500005],tov1[500005],tov2[500005];
int nex1[500005],nex2[500005],low[500005],dfn[500005];
int judge[500005],beer[500005],liquor[500005],tot,dis[500005];
int stark[500005],step,dex,bcnt,belong[500005],ans;
int mon[500005],money[500005];
void add1(int a,int b)
{
    tov1[++tot]=b;
    nex1[tot]=head1[a];
    head1[a]=tot;
}
void add2(int a,int b)
{
    tov2[++tot]=b;
    nex2[tot]=head2[a];
    head2[a]=tot;
}
void tarjan(int k)
{
    stark[++step]=k;
    low[k]=dfn[k]=++dex;
    judge[k]=1;
    int t=head1[k];
    while(tov1[t])
    {
        if(!dfn[tov1[t]])
        {
            tarjan(tov1[t]);
            low[k]=min(low[k],low[tov1[t]]);
        }
        else if(judge[tov1[t]])
            low[k]=min(low[k],dfn[tov1[t]]);
        t=nex1[t];
    }
    if(low[k]==dfn[k])
    {
        int j=0;
        ++bcnt;
        while(j!=k)
        {
            j=stark[step--];
            judge[j]=0;
            belong[j]=bcnt;
            money[bcnt]+=mon[j];
            if(beer[j])
                liquor[bcnt]=1;
        }
    }
}
void spfa(int o)
{
    memset(judge,0,sizeof judge);
    queue<int>q;
    q.push(o);
    dis[o]=money[o];
    if(liquor[o])
        ans=dis[o];
    while(!q.empty())
    {
        int k=q.front();judge[k]=0;
        q.pop();int t=head2[k];
        while(tov2[t])
        {
            if(dis[tov2[t]]<dis[k]+money[tov2[t]])
            {
                dis[tov2[t]]=dis[k]+money[tov2[t]];
                if(!judge[tov2[t]])
                {
                    judge[tov2[t]]=1;
                    q.push(tov2[t]);
                }
                if(liquor[tov2[t]])
                    ans=max(ans,dis[tov2[t]]);
            }
            t=nex2[t];
        }
    }

}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add1(a,b);
    }
    tot=0;
    for(int i=1;i<=n;i++)
        scanf("%d",&mon[i]);
    int s,p;scanf("%d%d",&s,&p);
    for(int i=1;i<=p;i++)
    {
        int a;scanf("%d",&a);beer[a]=1;
    }
    for(int i=1;i<=n;i++)
        if(!dfn[i])
            tarjan(i);
    for(int i=1;i<=n;i++)
    {
        int t=head1[i];
        while(tov1[t])
        {
            if(belong[i]!=belong[tov1[t]])
                add2(belong[i],belong[tov1[t]]);
            t=nex1[t];
        }
    }
    spfa(belong[s]);
    printf("%d",ans);
}