#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。ios
C++代碼:spa
#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.net