hduoj 2544 最短路（SPFA）

hduoj 2544 最短路

Problem Description

在每年的校赛里，所有进入决赛的同学都会获得一件很漂亮的t-shirt。但是每当我们的工作人员把上百件的衣服从商店运回到赛场的时候，却是非常累的！所以现在他们想要寻找最短的从商店到赛场的路线，你可以帮助他们吗？

Input

输入包括多组数据。每组数据第一行是两个整数N、M（N<=100，M<=10000），N表示成都的大街上有几个路口，标号为1的路口是商店所在地，标号为N的路口是赛场所在地，M则表示在成都有几条路。N=M=0表示输入结束。接下来M行，每行包括3个整数A，B，C（1<=A,B<=N,1<=C<=1000）,表示在路口A与路口B之间有一条路，我们的工作人员需要C分钟的时间走过这条路。
输入保证至少存在1条商店到赛场的路线。

Output

对于每组输入，输出一行，表示工作人员从商店走到赛场的最短时间

Sample Input

2 1
1 2 3
3 3
1 2 5
2 3 5
3 1 2
0 0

Sample Output

3
2

思路：SPFA

#include<bits/stdc++.h>
using namespace std;

struct SPFA{
    using PII = pair<int,int>;
    int n;
    vector<vector<PII>> G;
    vector<int> dis;
    vector<bool> vis;
    
    void Init(int n){
        this->n = n;
        G.resize(n);
        dis.resize(n);
        vis.resize(n);
    }
    
    void add_Edge(int u, int v, int w){
        G[u].push_back( {v,w} );
    }
    
    vector<int> get_Dist(int s){
        for(int i = 0; i < n; i++) dis[i] = 2e9;
        queue<int> q;
        dis[s] = 0;
        vis[s] = true;
        q.push(s);
        while(q.size()){
            int u = q.front(); q.pop();
            vis[u] = false;
            for(auto t : G[u]){
                int v = t.first, d = t.second;
                if(dis[v] > dis[u] + d){
                    dis[v] = dis[u] + d;
                    if(!vis[v]){
                        q.push(v);
                        vis[v] = true;
                    }
                }
            }
        }
        return dis;
    }
};

int main(){
    int n,m;
    while( cin >> n >> m and (n || m) ){
        SPFA spfa;
        spfa.Init(n + 1);
        while(m--){
            int u,v,w;
            cin >> u >> v >> w;
            spfa.add_Edge(u,v,w);
            spfa.add_Edge(v,u,w);
        }
        auto ans = spfa.get_Dist(1);
        cout << ans[n] << endl;
    }
    return 0;
}