SPFA 邻接表模式

poj 1511

一个SPFA 题意是：在一个城市里，有n（1~1000000）个运输中心，有m条有向路连接着任意的两个运输中心。ACM组织（位于编号为1的运输中心）要派发p个雇佣员工早上前往这p个运输中心去发传单，晚上再让他们都回到组织中。问所有人所走路程的总和最短是多少？

#include <vector>
#include <iostream>
#include <string>
#include <map>
#include <stack>
#include <cstring>
#include <queue>
#include <list>
#include <cstdio>
#include <set>
#include <algorithm>
#include <cstdlib>
#include <cmath>
#include <iomanip>
#include <cctype>
#include <sstream>
#include <functional>
using namespace std;

#define LL long long
#define INF 1000000050
#define endl '\n'
#define me(x) memset(x,0,sizeof(x));
//const LL INF = 1e15;
const int maxn=1e3+5;
const int maxx=1e6+5;

int n,m;
struct node
{
    int e,w,next;
}E[maxx],G[maxx];
int head1[maxx],head2[maxx];
int d[maxx],inq[maxx];
LL sum;
void init()
{
    memset(head1,-1,sizeof(head1));
    memset(head2,-1,sizeof(head2));
    memset(E,0,sizeof(E));
    memset(G,0,sizeof(G));
}
void spfa1()
{
    queue<int >Q;
    memset(inq,0,sizeof(inq));
    for(int i=0;i<=n;i++)
        d[i]=INF;
    Q.push(1),d[1]=0,inq[1]=1;
    while(!Q.empty())
    {
        int now=Q.front();
        Q.pop();
        inq[now]=0;
        for(int i=head1[now];i!=-1;i=E[i].next)
        {
            int v=E[i].e;
            if(d[v]>d[now]+E[i].w)
            {
                d[v]=d[now]+E[i].w;
                if(inq[v]==1) continue;
                inq[v]=1;
                Q.push(v);
            }
        }
    }
    for(int i=1;i<=n;i++)
        sum+=d[i];
}
void spfa2()
{
    queue<int >Q;
    memset(inq,0,sizeof(inq));
    for(int i=0;i<=n;i++)
        d[i]=INF;
    Q.push(1),d[1]=0,inq[1]=1;
    while(!Q.empty())
    {
        int now=Q.front();
        Q.pop();
        inq[now]=0;
        for(int i=head2[now];i!=-1;i=G[i].next)
        {
            int v=G[i].e;
            if(d[v]>d[now]+G[i].w)
            {
                d[v]=d[now]+G[i].w;
                if(inq[v]==1) continue;
                inq[v]=1;
                Q.push(v);
            }
        }
    }
    for(int i=1;i<=n;i++)
        sum+=d[i];
}
void solve()
{
    sum=0;
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        E[i].e=y;
        E[i].next=head1[x];
        E[i].w=z;
        head1[x]=i;
        G[i].e=x;
        G[i].next=head2[y];
        G[i].w=z;
        head2[y]=i;

    }
    spfa1();spfa2();
    printf("%I64d\n",sum);
}
int main()
{
    int t;
    scanf("%d",&t);
    while (t--)
    {
        solve();
    }
}