题目链接:https://www.patest.cn/contests/pat-a-practise/1030
题目大意:给一个图,对于每条边给出边的长度(城市距离)和边的权重(旅行花销)。然后给出起点和终点,求出两点之间的最短路径,若有重复则取边的总权值(总花销)最小的那条。
解题思路:
- 本题目与另一道PAT 1003思路相同,都是使用Dijkstra最短路径算法。
- 区别是:要将路径输出。在求最短路径的时候,使用pre[]数组记录最短路径,pre[j]=current表示j前一个点为current。
代码如下:
#include <iostream>
#include <stack>
using namespace std;
int edge[501][501],cost[501][501],dist[501],wi[501],pre[501]={-1};
const int inf=10000000;
int main(int argc, char const *argv[])
{
int N,M,S,D;
cin>>N>>M>>S>>D;
for(int i=0;i<N;i++){
for(int j=0;j<N;j++){
edge[i][j]=inf;
cost[i][j]=inf;
}
dist[i]=inf;
}
int c1,c2,d,c;
for(int i=0;i<M;i++){//输入每条边的信息
cin>>c1>>c2>>d>>c;
edge[c1][c2]=edge[c2][c1]=d;
cost[c1][c2]=cost[c2][c1]=c;
}
int visit[501]={0};
dist[S]=0;
wi[S]=0;
for(int i=0;i<N;i++){
int current=-1,min=inf;
for(int j=0;j<N;j++){
if(!visit[j]&&dist[j]<min){
min=dist[j];
current=j;
}
}
if(current==-1)
break;
visit[current]=1;
for(int j=0;j<N;j++){
if(!visit[j]&&edge[current][j]!=inf){
if(dist[current]+edge[current][j]<dist[j]){//找到更短的路径
dist[j]=dist[current]+edge[current][j];//更新最短路径
wi[j]=wi[current]+cost[current][j];//并跟新花费
pre[j]=current;
}
else if(dist[current]+edge[current][j]==dist[j]){//长度相同
if(wi[current]+cost[current][j]<wi[j]){//但是花费更短
wi[j]=wi[current]+cost[current][j];//只更新最少花费
pre[j]=current;
}
}
}
}
}
cout<<S;
int tmp=pre[D],k=0,path[501];//倒序的路径
while(tmp!=S){
path[k++]=tmp;
tmp=pre[tmp];
}
for(int i=k-1;i>=0;i--)
cout<<" "<<path[i];
cout<<" "<<D<<" "<<dist[D]<<" "<<wi[D]<<endl;
return 0;
}