POJ 1511 Invitation Cards

题意：有T组测试数据，每组测试数据有N个点，M个路线。接下来的M行有三个数字，分别代表每条路线的起点，终点，权值，从点1到其他点的和为A，其他点到点1的和为B，问A＋B的最小值。

SPFA

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int N=1000005;
bool vis[N];
int next1[N],next2[N],dis[N];
struct node
{
    int x,y,w,next;
    void fun(int a,int b,int c,int d){x=a;y=b;w=c;next=d;}
}edge1[N],edge2[N];
long long spfa(node*,int*,int);
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m,all1=1,all2=1;
        scanf("%d%d",&n,&m);
        memset(next1,0,sizeof(int)*(n+1));
        memset(next2,0,sizeof(int)*(n+1));
        for(int i=1;i<=m;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            edge1[all1].fun(a,b,c,next1[a]); next1[a]=all1++;
            edge2[all2].fun(b,a,c,next2[b]); next2[b]=all2++;
        }
        long long sum=spfa(edge1,next1,n)+spfa(edge2,next2,n);
        printf("%lld\n",sum);
    }
}
long long spfa(node *edge,int *next,int n)
{
    memset(vis,0,sizeof(vis));
    memset(dis,0x7f,sizeof(dis));

    dis[1]=0;vis[1]=1;
    queue<int> que;
    que.push(1);
    while(!que.empty())
    {
        int k=que.front(); que.pop();
        vis[k]=0;
        for(int i=next[k];i>0;i=edge[i].next)
        {
            int x=edge[i].x,y=edge[i].y,w=edge[i].w;
            if(dis[y]>dis[x]+w)
            {
                dis[y]=dis[x]+w;
                if(!vis[y])
                {
                    que.push(y);
                    vis[y]=1;
                }
            }
        }
    }
    long long sum=0;
    for(int i=2;i<=n;i++) sum+=dis[i];
    return sum;
}