Dijkstra算法求最短路径

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