bzoj 1179: [Apio2009]Atm spfa+强连通分量

题意

给一个有向图，给定起点，求一条路径（不一定是简单路径）使得经过的点的权值和最大（每个点只算一次），且要在给定的若干个点中选一个作为终点。
n,m<=500000

分析

我真的sb，居然没有一眼看出正解是啥。。。

只要缩点一下然后直接spfa就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
using namespace std;

typedef long long LL;

const int N=500005;

int cnt,ls[N],last[N],n,m,bel[N],tim,dfn[N],low[N],tot,s;
LL dis[N],a[N],val[N];
bool vis[N],ins[N];
queue<int> q;
stack<int> sta;
struct edge{int from,to,next;}e[N*2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge(int u,int v)
{
    e[++cnt].from=u;e[cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void addedge1(int u,int v)
{
    if (u==v) return;
    e[++cnt].to=v;e[cnt].next=ls[u];ls[u]=cnt;
}

void tarjan(int x)
{
    sta.push(x);ins[x]=1;
    dfn[x]=low[x]=++tim;
    for (int i=last[x];i;i=e[i].next)
        if (!dfn[e[i].to])
        {
            tarjan(e[i].to);
            low[x]=min(low[x],low[e[i].to]);
        }
        else if (ins[e[i].to]) low[x]=min(low[x],dfn[e[i].to]);
    if (low[x]==dfn[x])
    {
        tot++;int y=0;
        while (y!=x)
        {
            y=sta.top();sta.pop();
            ins[y]=0;bel[y]=tot;val[tot]+=a[y];
        }
    }
}

void spfa()
{
    dis[bel[s]]=val[bel[s]];q.push(bel[s]);vis[bel[s]]=1;
    while (!q.empty())
    {
        int u=q.front();q.pop();
        for (int i=ls[u];i;i=e[i].next)
            if (dis[u]+val[e[i].to]>dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+val[e[i].to];
                if (!vis[e[i].to]) vis[e[i].to]=1,q.push(e[i].to);
            }
        vis[u]=0;
    }
}

int main()
{
    n=read();m=read();
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read();
        addedge(x,y);
    }
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++) if (!dfn[i]) tarjan(i);
    int lim=cnt;
    for (int i=1;i<=lim;i++) addedge1(bel[e[i].from],bel[e[i].to]);
    s=read();
    spfa();
    int w=read();LL ans=0;
    while (w--)
    {
        int x=read();ans=max(ans,dis[bel[x]]);
    }
    printf("%lld",ans);
    return 0;
}