Dijkstra求最短路的條數,並輸出最短路徑和最短路經過的點的最大和
阿新 • • 發佈:2018-12-24
#include <cstdio> #include <algorithm> #include <iostream> #include <cstring> #include <stack> using namespace std; const int maxn=1e3; const int INF=1e8; int n,m,st,en,x,y,z; int val[maxn],valsum[maxn],path[maxn],pathsum[maxn];//val儲存每個點的值,valsum儲存到點【i】前的點值和,path儲存點【i】前那個點的號,pathsum儲存到點【i】的最短路的條數 int d[maxn],vis[maxn],G[maxn][maxn]; void dijkstra() { for(int i=0;i<n;i++) d[i]=G[st][i]; d[st]=0; pathsum[st]=1;//初始化起點的最短路的條數是1 int min1,p; for(int i=1; i<=n; i++) { min1=INF; for(int j=0; j<n; j++) { if(!vis[j] && d[j]<min1) { min1=d[j]; p=j; } } vis[p]=1; for(int j=0; j<n; j++) { if(!vis[j] && d[j]>d[p]+G[p][j]) { d[j]=d[p]+G[p][j]; pathsum[j]=pathsum[p];//當鬆弛時,到【j】和到【p】的最短路的條數相同 valsum[j]=valsum[p]+val[j];//鬆弛時直接加上更新後的點值 path[j]=p;//記錄上一個點 } else if(!vis[j] && d[j]==d[p]+G[p][j]) { pathsum[j]+=pathsum[p];//當相等時,到【j】的加上到【p】的 if(valsum[j]<valsum[p]+val[j]) { valsum[j]=valsum[p]+val[j];//當相等時加上大的那個 path[j]=p;//相等時記錄大的那個點 } } } } } void P()//輸出路徑 { int tmp; stack<int>si; while(!si.empty()) si.pop(); si.push(en);//從最後的那個點開始逆序存到堆疊裡 while (st!=en) { tmp=path[en]; si.push(tmp); en=tmp; } cout << si.top(); si.pop(); while(!si.empty()) { cout << ' ' << si.top(); si.pop(); } cout << endl; } void inition() { memset(path,0,sizeof(path)); memset(vis,0,sizeof(vis)); memset(pathsum,0,sizeof(pathsum)); for(int i=0; i<n; i++) { for(int j=0; j<n; j++) { if(i==j) G[i][j]=0; else G[i][j]=INF; } valsum[i]=val[i]; } } int main() { scanf("%d%d%d%d",&n,&m,&st,&en); for(int i=0; i<n; i++) scanf("%d",&val[i]); inition(); for(int i=1; i<=m; i++) {scanf("%d%d%d",&x,&y,&z);G[x][y]=z;G[y][x]=z;}//無向圖 dijkstra(); printf("%d %d\n",pathsum[en],valsum[en]); P(); return 0; }