(Dijkstra 输出路径) HDU 1874 畅通工程续
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),分别代表起点和终点。
每组数据第一行包含两个正整数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> 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 = s; for(int j = 0; j < n; j++){ if(vis[j] == 0 && dis[j] < minn){ minn = dis[j]; u = j; } } if(u == s) break; //指的是当没有了剩余的节点的话,就break。 也可以不用加上,也能过。 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){ 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; } } 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