JZOJ 4771. 爬山

Problem

Description

国家一级爬山运动员h10今天获得了一张有着密密麻麻标记的地图，在好奇心的驱使下，他又踏上了去爬山的路。
对于爬山，h10有一个原则，那就是不走回头路，于是他把地图上的所有边都标记成了有向边。他决定从点S出发，每到达一个新的节点他就可以获得一定的成就值。同时h10又是一个很珍惜时间的运动员，他不希望这次爬山的成就值白白浪费，所以最后他一定要在一个存档点停下，保存自己的成就值。
请你计算出在此次爬山运动中h10能够得到的最大成就值。保证h10能走到存档点。

Input

第一行两个整数 N，M，表示点数和边数。
接下来 M 行，每行两个整数 u，v，表示u到v有一条有向边(没有自环)。
第 M+2 行 N 个正整数，表示每个点的成就值。
接下来一行两个整数 S，p，表示出发点和存档点个数。
下面一行 p 个整数，表示存档点。

Output

一个正整数，表示最大成就值。

Sample Input

5 7
5 1
3 1
2 5
3 5
4 3
4 2
4 5
7 6 3 2 2
4 3
1 5 2

Sample Output

17

Data Constraint

对于 30% 的数据， N,M≤1000，并且地图为有向无环图。
对于 100% 的数据， N,M≤500000。（数据有梯度，注意答案的大小）

Solution

我们用Tarjan找到所有强连通分量，并将每个强连通分量缩成一个点，因为强连通分量中所有的分数都可以获得。
然后再跑一遍Spfa最长路，然后取最高的存档点的值即可。
然而，链型数据可能会卡爆你的Tarjan!
所以我们要打人工栈。人工栈的原理与递归的原理相同，不过就是要再开一个数组。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define N 500010
#define LL long long
using namespace std;
int n,m,i,j,h,t,w,top,cnt,x,y,p,S,now,tot;
int g[N],low[N],dfn[N],st[N],head[N],next[N],go[N],A[N];
LL a[N],d[N],dis[N],f[N],ans;
bool b[N],c[N],e[N],bz[N];
inline int read()
{
    int data=0;
    char ch=0;
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();
    return data;
}
void lb(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
void tarjan(int x)
{
    int T=1,i,now;bool o;
    A[T]=x;
    while (T)
    {
        if (dfn[A[T]]==-1)
        {
            h++;
            dfn[A[T]]=low[A[T]]=h;
            bz[A[T]]=1;
            st[++top]=A[T];
        }
        o=0;
        for (i=head[A[T]];i;i=next[i])
        {
            now=go[i];
            if (dfn[now]==-1)
            {
                o=1;
                break;
            } else
            if (bz[now]) low[A[T]]=min(low[A[T]],dfn[now]);
        }
        if (o) A[++T]=now;else
        {
            if (dfn[A[T]]==low[A[T]])
            {
                cnt++;
                while (st[top+1]!=A[T])
                {
                    bz[st[top]]=0;
                    g[st[top]]=cnt;
                    if (c[st[top]]) e[cnt]=1;
                    f[cnt]+=a[st[top]];
                    top--;
                }
            }
            if (T>1) low[A[T-1]]=min(low[A[T-1]],low[A[T]]);
            T--;
        }
    }
}
int main()
{
    memset(dfn,255,sizeof(dfn));
    n=read();m=read();
    fo(i,1,m) x=read(),y=read(),lb(x,y);
    fo(i,1,n) a[i]=read();
    scanf("%d%d",&S,&p);
    fo(i,1,p) x=read(),c[x]=1;
    tot=0;
    tarjan(S);
    t=0,w=1;
    d[1]=S;
    dis[S]=f[g[S]];
    while (t<w)
    {
        t++;
        now=d[t];
        for(i=head[now];i;i=next[i])
        {
            if (g[go[i]]!=g[now])
            {
                if (dis[now]+f[g[go[i]]]>dis[go[i]])
                {
                    dis[go[i]]=dis[now]+f[g[go[i]]];
                    if (!b[go[i]])
                    {
                        b[go[i]]=1;
                        w++;
                        d[w]=go[i];
                    }
                }
            } else
            {
                if (dis[now]>dis[go[i]])
                {
                    dis[go[i]]=dis[now];
                    if (!b[go[i]])
                    {
                        b[go[i]]=1;
                        w++;
                        d[w]=go[i];
                    }
                }
            }
        }
        b[now]=0;
    }
    fo(i,1,n) if (c[i]) ans=max(ans,dis[i]);
    printf("%lld",ans);
}