算法原理:
此算法适用于单源最短路径求解。
1.初始化:集合final<bool>为已经求解了最短路径的顶点集,初始时,将final[源点]设为true,其他都为false。集合dist[j]为源点到j的最短距离,集合path[j]为当前j走到顶点j的上一个顶点。
2.每次从未加入final集合的顶点中选取一个到已求解集合中距离最短的顶点,加入集合(将final设为true)。然后修改path[cur]=pre;最后更新源点到其他所有点的距离dist[],看:经过当前点到其他所有点的距离小,还是不经过当前点到其他点的距离更小,选取二者较小值;
使用邻接矩阵存储图时,该算法时间复杂度为O(n^2);
#include<iostream>
#include<vector>
#define INF INTMAX_MAX
using namespace std;
void Dijkstra(int n,int graph[][5]){
vector<int>dist(n,0);
vector<bool>final(n,false);
vector<int>path(n,0);
for(int i=0;i<n;i++){
dist[i]=graph[0][i];
}
final[0]=true;
int pre=0;
int cur=0;
for(int i=0;i<n;i++){
int min=INF;
for(int j=0;j<n;j++){
if(!final[j]&&min>dist[j]){
min=dist[j];
cur=j;
}
}
final[cur]=true;
path[cur]=pre;
pre=cur;
for(int j=0;j<n;j++){
dist[j]=dist[j]<(dist[cur]+graph[cur][j])?dist[j]:(dist[cur]+graph[cur][j]);
}
}
for(int i=0;i<n;i++){
cout<<"到达"<<i<<"点的最小距离"<<dist[i]<<endl;
}
}
int main(){
int graph[5][5]={
{0,10,INF,INF,5},
{INF,0,1,INF,2},
{INF,INF,0,4,INF},
{2,INF,6,0,INF},
{INF,3,9,2,0}
};
Dijkstra(5,graph);
}