題目描述
給出一個圖的鄰接矩陣,輸入頂點v,用迪傑斯特拉算法求頂點v到其它頂點的最短路徑。
輸入
第一行輸入t,表示有t個測試實例
第二行輸入頂點數n和n個頂點信息
第三行起,每行輸入鄰接矩陣的一行,以此類推輸入n行
第i個結點與其它結點如果相連則為距離,無連接則為0,數據之間用空格
隔開。第四行輸入v0,表示求v0到其他頂點的最短路徑距離
以此類推輸入下一個示例
輸出
對每組測試數據,輸出:
每行輸出v0到某個頂點的最短距離和最短路徑
每行格式:v0編號-其他頂點編號-最短路徑值----[最短路徑]。沒有路徑輸出:v0編號-其他頂點編號--1。具體請參考示范數據
樣例輸入
2 5 0 1 2 3 4 0 5 0 7 15 0 0 5 0 0 0 0 0 0 1 0 0 2 0 0 0 0 0 0 0 0 6 V0 V1 V2 V3 V4 V5 0 0 10 0 30 100 0 0 5 0 0 0 0 0 0 50 0 0 0 0 0 0 0 10 0 0 0 20 0 60 0 0 0 0 0 0 V0
樣例輸出
0-1-5----[0 1 ] 0-2-9----[0 3 2 ] 0-3-7----[0 3 ] 0-4-10----[0 3 2 4 ] V0-V1--1 V0-V2-10----[V0 V2 ] V0-V3-50----[V0 V4 V3 ] V0-V4-30----[V0 V4 ] V0-V5-60----[V0 V4 V3 V5 ]
提示
#include<iostream> #include<stack> #include<cstring> using namespace std; #define INF 0x3f3f3f3f int m[50][50]; int process[50][50]; int n; int find(string *str,string s) { for(int i=0;i<n;i++) if(str[i]==s) return i; return -1; } int minindex(int j,int tag[50]) { int index=0; for(int i=0;i<n;i++) { if(tag[i]==0&&process[i][j]<process[index][j]) index=i; } return index; } void DIJ(string *str,int start) { int path[50]={0}; for(int i=0;i<50;i++) for(int j=0;j<50;j++) process[i][j]=INF; int tag[50]={0}; tag[start]=1; path[start]=-1; for(int j=0;j<n;j++) { if(tag[j]==0) process[j][0]=m[start][j]; }///第0列初始化 int vmin=minindex(0,tag); tag[vmin]=1; for(int j=1;j<n;j++)///列從1到n-1 { for(int i=0;i<n;i++)///行從0到n-1 { process[i][j]=process[i][j-1];///先把前一列復制過來 } for(int i=0;i<n;i++) { if(tag[i]==0&&m[start][vmin]+m[vmin][i]<m[start][i]) {///能優化就把process和m的值都對應改變 process[i][j]=m[start][vmin]+m[vmin][i]; m[start][i]=process[i][j]; path[i]=vmin; } } vmin=minindex(j,tag);///挑出下一個vmin,即start到vmin此時最短 tag[vmin]=1;///vmin標記已走 } for(int i=0;i<n;i++) { stack<int>Q; if(i==start) continue; cout<<str[start]<<"-"<<str[i]<<"-"; if(process[i][n-1]>0x3f3f3f) cout<<"-1"<<endl; else { cout<<process[i][n-1]<<"----["; int j=path[i]; while(j!=-1) { Q.push(j); j=path[j]; } while(!Q.empty()) { cout<<str[Q.top()]<<" "; Q.pop(); } cout<<str[i]<<" "<<"]"<<endl; } } } int main() { int T; cin>>T; while(T--) { memset(m,0,sizeof(m)); cin>>n; string *str=new string[n]; for(int i=0;i<n;i++) cin>>str[i]; for(int i=0;i<n;i++) for(int j=0;j<n;j++) { cin>>m[i][j]; if(m[i][j]==0) m[i][j]=INF; } string start; cin>>start; int v=find(str,start); DIJ(str,v); delete []str; } return 0; }
