Invitation Cards POJ-1511 (spfa)

题目链接：Invitation Cards

题意：

　　给出一张有向图，现在要求从1到其他所有的结点的最小路径和与从所有其他结点到1的最小路径和之和。

题解：

　　求最小路径可以用SPFA来求解。从1到其他结点的正向跑一遍SPFA就可以求出来了，要求其他结点到1的最小路径则可以反向建图跑一边SPFA。但是这里面很奇怪的是，如果正向和反向建图开两个vector就会TLE，开一个就不会TLE了。@。@？

#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef pair<int,int> P;
struct edge{
    int u,v,val;
};

const int MAX_N = 1e6+9;
const int INF = 1e9+7;
vector<P> vec[MAX_N];
vector<P> vec2[MAX_N];
int res[MAX_N];
int vis[MAX_N];
edge tran[MAX_N];
queue<int> que;
int N,M,T,a,b,x;
long long ans;
void init()
{
    for(int i=0;i<=N;i++)
    {
        vec[i].clear();
        vec2[i].clear();
    }
    while(!que.empty()) que.pop();
    ans = 0;
}
void spfa(int z)
{
    while(!que.empty()) que.pop();
    for(int i=0;i<=N;i++)
    {
        res[i] = INF;
        vis[i] = 0;
    }
    res[z] = 0;
    vis[z] = 1;
    que.push(z);
    while(!que.empty())
    {
        int t = que.front();
        que.pop();
        vis[t] = 0;
        for(int i=0;i<vec[t].size();i++)
        {
            P temp = vec[t][i];

            if(res[temp.first] > res[t] + temp.second)
            {
                res[temp.first] = res[t] + temp.second;
                if(vis[temp.first] == 0)
                {
                    vis[temp.first] = 1;
                    que.push(temp.first);
                }
            }

        }
    }
    for(int i=2;i<=N;i++) ans += res[i];
}

int main()
{
    cin>>T;
    while(T--)
    {
        scanf("%d%d",&N,&M);
        init();
        for(int i=0;i<M;i++)
        {
            scanf("%d%d%d",&tran[i].u,&tran[i].v,&tran[i].val);
            vec[tran[i].u].push_back(P(tran[i].v,tran[i].val));
        }
        spfa(1);
        for(int i=0;i<=N;i++) vec[i].clear();
        for(int i=0;i<M;i++)
        {
            vec[tran[i].v].push_back(P(tran[i].u,tran[i].val));
        }
        spfa(1);
        printf("%lld
",ans);
    }
    return 0;
}