ATM+bzoj+Trajan算法+spfa求最短路

題目鏈接
在新的bzoj平臺上，目前這道題沒有ac的人。我的代碼也卡在了12，13的樣例上，在網上找了很多的代碼，除了一位大佬（卡在12上），其餘都是卡在了12，13上。

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int num1=5100010;
const int num2=5100010;
/*--------------------*/
struct node
{
    int u,v,next;
    LL w;
    node(){};
    node(int x,LL y){
        v=x;w=y;
    };
    bool operator < (const node & a)const{
        return w < a.w;//求最短路時，是w>a.w;求最長路，相反；
    }
}e[num2];
int head[num1],cnt,cnt2;
/*--------------------*/
int dfn[num1],low[num1],ind,stack[num1],t,cnt3,f[num1];
LL w[num1],w2[num1];
bool vis[num1];
/*--------------------*/
int n,m,s,p;
LL ans;
int b[num1];
/*--------------------*/
LL dis[num1];
/*------------------*/
void int_i(void)
{
    cnt=-1;
    memset(head,-1,sizeof(head));
    return ;
}
void addedge(int u,int v,LL w=0)
{
    e[++cnt].u=u;
    e[cnt].v=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
    return;
}
void tarjan(int u)
{
    int v;
    dfn[u]=low[u]=++ind;
    stack[++t]=u;
    vis[u]=true;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        v=e[i].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(vis[v]){
            low[u]=min(low[u],dfn[v]);
        }
    }

    if(dfn[u]==low[u])
    {
        cnt3++;
        do{
            v=stack[t--];
            vis[v]=false;
            f[v]=cnt3;
            w2[cnt3]+=w[v];
        }while(v!=u);
    }
    return ;
}
void spfa(int s)
{
    bool book[num1];
    for(int i=1;i<=cnt3;i++)
    {
        dis[i]=0;
        book[i]=false;
    }
    dis[s]=w2[s];
    queue<int>q;
    q.push(s);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        book[x]=0;
        for(int i=head[x];i!=-1;i=e[i].next)
        {
            int y=e[i].v;
            if(dis[y]<dis[x]+e[i].w)
            {
                dis[y]=dis[x]+e[i].w;
                if(!book[y])
                {
                    book[y]=1;
                    q.push(y);
                }
            }
        }
    }
}
int main()
{
    int u,v;
    int_i();
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&w[i]);
    }

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

    cnt2=cnt;
    memset(head,-1,sizeof(head));
    cnt=-1;
    for(int i=0;i<=cnt2;i++)
    {
        u=e[i].u;
        v=e[i].v;
        if(f[u]!=f[v])
        {
            u=f[u];v=f[v];
            addedge(u,v,w2[v]);//點權變邊權；
        }
    }

    scanf("%d%d",&s,&p);
    spfa(f[s]);
    for(int i=1;i<=p;i++)
    {
        scanf("%d",&b[i]);
        b[i]=f[b[i]];
        ans=max(ans,dis[b[i]]);
    }
    printf("%lld\n",ans);
    return 0;
}