Dijkstra算法求单源最短路径
算法思想:贪心算法
//
// Created by admin on 2022/10/31.
//
#include <iostream>
#include <algorithm>
#include <stack>
using namespace std;
const int N=100; //足够大的定点数量
const int INF=1e7;
int map[N][N]; // 边的集合
int dist[N]; // 源点到各个点的距离的集合
bool flag[N];
int n,m; // n表示顶点数 m表示边的数量
void Dijkstra(int u){ //u为源点
dist[u]=0;// 表示源点u到u的距离
flag[u]= true;
for(int i=1;i<=n;i++){
int temp=INF;
int t=u;
for(int j=1;j<=n;j++){
if(!flag[j]&&dist[j]<temp){
t=j;
temp=dist[j];
}
}
flag[t]= true;
for(int j=1;j<=n;j++){
if(!flag[j]&&map[t][j]<INF){
if(dist[j]>dist[t]+map[t][j]){
dist[j]=dist[t]+map[t][j];
}
}
}
}
}
int main(){
int begin, end, length;
int u;
cout<<"请输入顶点个数:"<<endl;
cin>>n;
cout<<"请输入边的数量:"<<endl;
cin>>m;
for(int i=1;i<=m;i++){
for(int j=1;j<=m;j++){
map[i][j]=INF;
}
}
cout<<"请输入边"<<endl;
for(int i=1;i<=m;i++){
cin>>begin>>end>>length;
map[begin][end]=length;
}
cout<<"请输入源点"<<endl;
cin>>u;
for(int i=1;i<=n;i++){
dist[i]=map[u][i];
flag[i]= false;
}
Dijkstra(u);
for(int i=1;i<=n;i++){
cout<<u<<"---->"<<i<<"="<<dist[i]<<endl;
}
}
测试
请输入顶点个数:
5
请输入边的数量:
8
请输入边
1 2 2
1 3 5
2 3 2
2 4 6
3 4 7
4 3 2
3 5 1
4 5 4
请输入源点
1
1---->1=0
1---->2=2
1---->3=4
1---->4=8
1---->5=5