最短路——逆向构图

题目：

Invitation Cards 题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1535

题目大意：

1、从起点到各个点的最短路。2、从各个终点到起点的最短路。

分析：

本题如果对于每个点都跑一次最短路，必定会超时，因此想到逆向构图法, 求各个点返回的最短路也就是在逆向图求起点到各个点的最短路。

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 10;
const int inf = 1e9;
struct node
{
    int y;
    int z;
};
int N, M;
bool v[maxn];
int d[maxn];
int sum;

void spfa(vector<node> vec[], queue<int> q, int N)
{
    int i;
    for(i = 1; i <= N; i++)
        d[i] = i == 1 ? 0 : inf;
    memset(v, false, sizeof(v));
    q.push(1), v[1] = true;
    while(!q.empty())
    {
        int x = q.front();
        q.pop(), v[x] = false;

        for(i = 0; i < vec[x].size(); i++)
            if(d[x] + vec[x][i].z < d[vec[x][i].y])
            {
                d[vec[x][i].y] = d[x] + vec[x][i].z;
                if(!v[vec[x][i].y])
                    v[vec[x][i].y] = true, q.push(vec[x][i].y);
            }
    }
    for(i = 2; i <= N; i++)
        sum += d[i];
}
int main()
{
    int T;
    int i, j, k, s;

    cin >> T;
    while(T--)
    {
        scanf("%d%d", &N, &M);
        queue<int> q1;
        queue<int> q2;
        vector<node> vec1[N + 1];
        vector<node> vec2[N + 1];

        sum = 0;
        for(i = 1; i <= M; i++)
        {
            int x, y, z;

            scanf("%d%d%d", &x, &y, &z);
            vec1[x].push_back((node){y, z});
            vec2[y].push_back((node){x, z});
        }
        spfa(vec1, q1, N);
        spfa(vec2, q2, N);
        printf("%d\n", sum);
    }
    return 0;
}