POJ-1511 Invitation Cards

题目大意：

给你p个人，有q条路径，这q条路径是有向的，问你从1号点走到其它各个点再走回来的cost总和是多少

解题思路：

因为p和q非常大， 都是1e6的数据大小，可以选择两种思路，

一个是dijkstra+heap优化可以达到O(nlogn)的时间复杂度

另外一个是spfa可以达到O(ke)的时间复杂度其中k<=2

这里采用了spfa算法

代码：

//memery time
//70708K 5063MS
#include <queue>
#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
#define INF 0x7fffffff
const int maxn = 1e6 + 5;

typedef long long LL;
typedef struct node{
    int to, w;
    node(int a = 0, int b = 0){
        to = a; w = b;
    }
}Edge;

LL dis[maxn];
int vis[maxn];
vector<Edge> vec[maxn];
int p, q, a[maxn], b[maxn], c[maxn];

LL spfa(int s){
    Edge v;
    queue<int> que;
    LL tmp, len, res = 0;
    while(!que.empty()) que.pop();
    for(int i = 0; i <= p; ++i){
        dis[i] = INF;
        vis[i] = 0;
    }
    dis[s] = 0;
    vis[s] = 1;
    que.push(s);

    while(!que.empty()){
        tmp = que.front();
        que.pop();

        len = vec[tmp].size();
        for(int i = 0; i < len; ++i){
            v = vec[tmp][i];
            if(dis[tmp] + v.w < dis[v.to]){
                dis[v.to] = dis[tmp] + v.w;
                if(!vis[v.to]){
                    vis[v.to] = 1;
                    que.push(v.to);
                }
            }
        }
        vis[tmp] = 0;
    }

    for(int i = 2; i <= p; ++i) res += dis[i];
    return res;
}
int main(){
    int t;
    scanf("%d", &t);
    while(t--){
        scanf("%d%d", &p, &q);
        for(int i = 0; i <= p; ++i)
            vec[i].clear();
        for(int i = 0; i < q; ++i){
            scanf("%d%d%d", &a[i], &b[i], &c[i]);
            vec[a[i]].push_back(node(b[i], c[i]));
        }

        LL ans = spfa(1);
        for(int i = 0; i <= p; ++i) vec[i].clear();
        for(int i = 0; i < q; ++i){
            vec[b[i]].push_back(node(a[i], c[i]));
        }
        ans += spfa(1);
        printf("%I64d
", ans);
    }
    return 0;
}