HDU 1535 Invitation Cards(有向图单源最短路径+SPFA)

第一次用SPFA求单源最短路径。

题意:求1到各点的最短路径和各点到1的最短路径之和。

先正向建图，用SFPA求出1到各点的最短路径；

再反向建图，用SPFA再求一次.(真实情况只要沿着路径反向走即是该点到1点的最短路径)

#include <cstdio>
#include <queue>
#include <iostream>
using namespace std;
typedef long long ll;
#define inf 99999999
const int maxn=1000000+5;
struct node
{
    int to,val;
};
int n;
ll dist[maxn];
vector<node>map1[maxn];
vector<node>map2[maxn];
void SPFA(vector<node>map[])
{
    for(int i=2;i<=n;i++) dist[i]=inf;
    dist[1]=0;
    queue<int>q;
    q.push(1);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        for(int i=0;i<map[u].size();i++)
        {
            int to=map[u][i].to;
            int val=map[u][i].val;
            if(dist[to]>dist[u]+val){
                dist[to]=dist[u]+val;
                q.push(to);
            }
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)   map1[i].clear(),map2[i].clear();
        while(m--)
        {
            node p1,p2;
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            p1.to=v;p1.val=w;
            p2.to=u;p2.val=w;
            map1[u].push_back(p1);
            map2[v].push_back(p2);
        }
        ll ans=0;
        SPFA(map1);
        for(int i=2;i<=n;i++) ans+=dist[i];
        SPFA(map2);
        for(int i=2;i<=n;i++) ans+=dist[i];
        printf("%lld\n",ans);
    }
    return 0;
}