pku1511 Invitation Cards

http://poj.org/problem?id=1511

图论，最短路，SPFA

求所有点到1

#include <stdio.h>
#include <string.h>
#include <queue>
#define N 1000100

using namespace std;

int n, m, src;
int x[N], y[N], len[N];
int dist[N];
bool inQue[N];
queue<int> que;
const int inf = 1<<30;
vector<pair<int, int> > g[N];

void spfa()
{
    int i;
    for(i=0; i<N; i++)
    {
        inQue[i] = false;
        dist[i] = inf;
    }
    dist[src] = 0;
    while(!que.empty())
    {
        que.pop();
    }
    que.push(src);
    inQue[src] = true;
    while(!que.empty())
    {
        int u = que.front();
        que.pop();
        for(i=0; i<g[u].size(); i++)
        {
            if(dist[u] + g[u][i].second < dist[g[u][i].first])
            {
                dist[g[u][i].first] = dist[u] + g[u][i].second;
                if(!inQue[g[u][i].first])
                {
                    inQue[g[u][i].first] = true;
                    que.push(g[u][i].first);
                }
            }
        }
        inQue[u] = false;
    }
}


int main()
{
    int t, i, j;
    long long result;
    scanf("%d", &t);
    while(t-- && scanf("%d%d", &n, &m))
    {
        for(i=1; i<=n; i++)
        {
            g[i].clear();
        }
        for(i=1; i<=m; i++)
        {
            scanf("%d%d%d", x+i, y+i, len+i);
        }
        for(i=1; i<=m; i++)
        {
            g[x[i]].push_back(make_pair(y[i], len[i]));
        }
        result = 0;
        src = 1;
        spfa();
        for(i=2; i<=n; i++)
        {
            result += dist[i];
        }
        for(i=1; i<=n; i++)
        {
            g[i].clear();
        }
        for(i=1; i<=m; i++)
        {
            g[y[i]].push_back(make_pair(x[i], len[i]));
        }
        src = 1;
        spfa();
        for(i=2; i<=n; i++)
        {
            result += dist[i];
        }
        printf("%lld\n", result);
    }
    return 0;
}