增加了一个记录顶点序列的数组,DFS输出(递归输出)
代码:
#include <stdio.h>
#include <algorithm>
using namespace std;
const int maxn = 1000;
const int INF = 100000000;
int n,m,s,G[maxn][maxn];
int d[maxn]; //记录路径
int pre[maxn]; //记录最短路径上每个顶点的前驱节点
bool vis[maxn] = {false};
void Dijkstra(int s){
fill(d,d+maxn,INF);
for(int i = 0;i <n;i++) pre[i] = i; //初始状态设每个点的前驱为自身
d[s] = 0;
for(int i = 0;i <n;i++){
int u = -1,MIN = INF;
for(int j = 0;j <n;j++){
if(vis[j] == false && d[j] <MIN){
u = j;
MIN = d[j];
}
}
if(u == -1) return;
vis[u] = true;
for(int v = 0;v <n;v++){
if(vis[v] == false && G[u][v] != INF && d[u]+G[u][v] <d[v]){
d[v] = d[u] + G[u][v];
pre[v] = u; //记录前驱节点
}
}
}
}
void DFS(int s,int v){
if(v == s){
printf("%d ",s);
return;
}
DFS(s,pre[v]);
printf("%d ",v);
}
int main(){
int u,v,w;
scanf("%d%d%d",&n,&m,&s);
fill(G[0],G[0]+maxn*maxn,INF);
for(int i = 0;i <m;i++){
scanf("%d%d%d",&u,&v,&w);
G[u][v] = w;
}
Dijkstra(s);
for(int i = 0;i <n;i++){
printf("顶点%d的最短距离为:%d\n",i,d[i]); //输出所有顶点的最短距离
printf("顶点%d的最短路径上顶点序列为:\n",i);
DFS(s,i); //每个顶点最短路径上的顶点编号
printf("\n");
}
}
/*
6 8 0
0 1 1
0 3 4
0 4 4
1 3 2
2 5 1
3 2 2
3 4 3
4 5 3
*/