(Dijkstra 输出路径) HDU 1784 畅通工程续

Posted 2021-12-02 weixu-liu

Problem Description

某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

 

 

Input

本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。

 

 

Output

对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1.

 

 

Sample Input

3 3

0 1 1

0 2 3

1 2 1

0 2

3 1

0 1 1

1 2

 

 

Sample Output

2

-1

 

 

这个是Dijkstra模板题。

C++代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<stack>
using namespace std;
const int maxn = 210;
const int INF = 0x3f3f3f3f;
int mp[maxn][maxn],vis[maxn],pre[maxn],dis[maxn],n,m,s,t;
void dijkstra(int s){
    for(int i = 0; i < n; i++){
        dis[i] = mp[s][i];
        vis[i] = 0;
        pre[i] = -1;
    }
    vis[s] = 1;
    dis[s] = 0;
    for(int i = 0; i < n; i++){
        int minn = INF,u = 0;
        for(int j = 0; j < n; j++){
            if(vis[j] == 0 && dis[j] < minn){
                minn = dis[j];
                u = j;
            }
        }
        vis[u] = 1;
        for(int j = 0; j < n; j++){
            if(vis[j] == 0 && dis[u] + mp[u][j] < dis[j]){
                dis[j] = dis[u] + mp[u][j];
                pre[j] = u;
            }
        }
    }
} 
int main(){
    while(cin>>n>>m){
        stack<int> St;
        for(int i = 0; i < n; i++){
            for(int j = 0;j < n; j++){
                mp[i][j] = INF;
            }
        }
        int u,v,w;
        for(int i = 0; i < m; i++){
            cin>>u>>v>>w;
            if(w < mp[u][v])   //这个必须加上，这个无向图中两个点之间可能会有多条路，求最短的就行。还有，由于是无向图，所以mp[u][v] = mp[v][u] = w得都写上。
                mp[u][v] = mp[v][u] = w;  
        }
        cin>>s>>t;
        dijkstra(s);
        if(dis[t] == INF){
            cout<<-1<<endl;
        }
        else{
            cout<<dis[t]<<endl;

        }    
        cout<<endl;
    }
    return 0;
}

 

 现在我特意进行拓展，打算使其能够输出最短路径，可以用stack。

C++代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<stack>
using namespace std;
const int maxn = 210;
const int INF = 0x3f3f3f3f;
int mp[maxn][maxn],vis[maxn],pre[maxn],dis[maxn],n,m,s,t;
void dijkstra(int s){
    for(int i = 0; i < n; i++){
        dis[i] = mp[s][i];
        vis[i] = 0;
        pre[i] = -1;
    }
    vis[s] = 1;
    dis[s] = 0;
    for(int i = 0; i < n; i++){
        int minn = INF,u = 0;
        for(int j = 0; j < n; j++){
            if(vis[j] == 0 && dis[j] < minn){
                minn = dis[j];
                u = j;
            }
        }
        vis[u] = 1;
        for(int j = 0; j < n; j++){
            if(vis[j] == 0 && dis[u] + mp[u][j] < dis[j]){
                dis[j] = dis[u] + mp[u][j];
                pre[j] = u;
            }
        }
    }
} 
int main(){
    while(cin>>n>>m){
        stack<int> St;
        for(int i = 0; i < n; i++){
            for(int j = 0;j < n; j++){
                mp[i][j] = INF;
            }
        }
        int u,v,w;
        for(int i = 0; i < m; i++){
            cin>>u>>v>>w;
            if(w < mp[u][v])
                mp[u][v] = mp[v][u] = w;
        }
        cin>>s>>t;
        dijkstra(s);
        if(dis[t] == INF){
            cout<<-1<<endl;
        }
        else{
            cout<<dis[t]<<endl;
            printf("输出路径：\n");
            int p = t;
            while(pre[p] != -1){
                St.push(p);
                p = pre[p];
            }
            St.push(p);
            cout<<s<<" ";     
            while(!St.empty()){
                int a = St.top();
                cout<<a<<" ";
                St.pop();
            }
        }    
        cout<<endl;
    }
    return 0;
}

 

参考链接：https://blog.csdn.net/renzijing/article/details/80572549