HDU SPFA算法 Invitation Cards

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

分析：

         题意：求1点到其它点的最短距离之和+其它点到1点的最短距离之和

         前面一部分直接用SPFA算法求出，而后一部分可用一数组存放反向边

       （所有边的方向都反一下），利用反向边SPFA求出1点到其它点距离即可。

 

#include <iostream>

#include <cstdio>

#include <cmath>

#include <cstdlib>

#include <string>

#include <cstring>

#include <algorithm>

#include <queue>



using namespace std;

const int inf = 0xfffffff;

const int maxn = 1000000+10;



bool vis[maxn];

int h1[maxn],h2[maxn],dis[maxn];

int t1,t2,n,m;



struct node{

    int x,v,next;

}f1[maxn],f2[maxn];

///f1存放顺向边， f2存放反向边



void init(){

    t1=t2=0;

    memset(h1,-1,sizeof(h1));

    memset(h2,-1,sizeof(h2));

}

void addnode_1(int a,int b,int c){

    f1[t1].x=b;

    f1[t1].v=c;

    f1[t1].next=h1[a];

    h1[a]=t1++;

}

void addnode_2(int a,int b,int c){

    f2[t2].x=b;

    f2[t2].v=c;

    f2[t2].next=h2[a];

    h2[a]=t2++;

}



int spfa(node F[ ],int H[ ]){

    memset(vis,false,sizeof(vis));

    for(int i=1;i<=n;++i)

        dis[i]=inf;

    dis[1]=0;

    vis[1]=true;

    queue<int>M;

    M.push(1);

    while(!M.empty()){

        int now=M.front(); M.pop();

        vis[now]=false;

        for(int i=H[now];i!=-1;i=F[i].next){

            int next=F[i].x;

            if(dis[next]>dis[now]+F[i].v){

                dis[next]=dis[now]+F[i].v;

                if(!vis[next]){

                    vis[next]=true;

                    M.push(next);

                }

            }

        }

    }

    int sum=0;

    for(int i=2;i<=n;++i)

        sum+=dis[i];

    return sum;

}



int main(){

    int T;  scanf("%d",&T);

    while(T--){

        scanf("%d%d",&n,&m);

        init();

        while(m--){

            int a,b,c;

            scanf("%d%d%d",&a,&b,&c);

            addnode_1(a,b,c);

            addnode_2(b,a,c);

        }

        int ans=spfa(f1,h1)+spfa(f2,h2);

        cout<<ans<<endl;

    }

    return 0;

}