zoj 2008 单源最短路 SPFA

/*
题意：一个有向带权图，求1到所有点的最小权值，再求所有点到1的最小权值

题解：典型单源最短路，用SPFA实现，要求1分别为起点和终点时的最小权值和，
当1为终点时只需将图建立一个反向图，即可将终点当作起点计算
*/
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>

#define Max 1000000001

using namespace std;

typedef struct
{
    int pr;
    int to;
}node;

int p;

vector<node> G[1000005],G0[1000005];
int dist[1000005];
bool vis[1000005];

void spfa(int start)
{
    for(int i=0; i<=p; i++)
    {
        dist[i] = Max;
        vis[i] = false;
    }
    queue<int> Q;
    dist[start] = 0;
    vis[start] = true;
    Q.push(start);
    while (!Q.empty())
    {
        int t = Q.front();
        Q.pop();
        for(int i=0; i<G[t].size(); i++)
        {
            if (dist[t]+G[t][i].pr < dist[G[t][i].to])
            {
                dist[G[t][i].to] = dist[t] + G[t][i].pr;
                if (!vis[G[t][i].to])
                {
                    Q.push(G[t][i].to);
                    vis[G[t][i].to] = true;
                }
            }
        }
        vis[t] = false;
    }
}

void spfa0(int start)
{
    for(int i=0; i<=p; i++)
    {
        dist[i] = Max;
        vis[i] = false;
    }
    queue<int> Q;
    dist[start] = 0;
    vis[start] = true;
    Q.push(start);
    while (!Q.empty())
    {
        int t = Q.front();
        Q.pop();
        for(int i=0; i<G0[t].size(); i++)
        {
            if (dist[t]+G0[t][i].pr < dist[G0[t][i].to])
            {
                dist[G0[t][i].to] = dist[t] + G0[t][i].pr;
                if (!vis[G0[t][i].to])
                {
                    Q.push(G0[t][i].to);
                    vis[G0[t][i].to] = true;
                }
            }
        }
        vis[t] = false;
    }
}

int main(void)
{
    int n,q,fr,to,price;
    cin >> n;
    while (n--)
    {
        cin >> p >> q;
        for(int i=0; i<=p; i++)
        {
            G[i].clear();
            G0[i].clear();
        }
        while (q--)
        {
            cin >> fr >> to >> price;
            node tmp;
            tmp.to = to;
            tmp.pr = price;
            G[fr].push_back(tmp);
            tmp.to = fr;
            G0[to].push_back(tmp);
        }
        spfa(1);
        int sum = 0;
        for(int i=1; i<=p; i++)
            sum += dist[i];
        spfa0(1);
        for(int i=1; i<=p; i++)
            sum += dist[i];
        cout << sum << endl;
    }
    return 0;
}