浙大数据结构习题笔记：Dijkstra算法（有权图最短路）

Dijkstra算法（有权图最短路）

具体教程内容见慕课
具体代码为可执行版，按有权图模式规范化输入即可，输出第一行为各行dict代表某结点从原点过来的总权重，输出第二行path为上一个结点的路径

具体代码实现

#include<iostream>
#include<stdlib.h>
#define Inf 1000000
#define Init -1
#define MaxVertex 100
typedef int Vertex;
int G[MaxVertex][MaxVertex];
int dist[MaxVertex];  // 距离
int path[MaxVertex];  // 路径
int collected[MaxVertex];  // 被收录集合
int Nv;   // 顶点
int Ne;   // 边
using namespace std;

// 初始化图信息
void build(){
    Vertex v1,v2;
    int w;
    printf("Enter Nv:");
    cin>>Nv;
    // 初始化图
    for(int i=1;i<=Nv;i++)
        for(int j=1;j<=Nv;j++)
            G[i][j] = 0;
    // 初始化路径
    for(int i=1;i<=Nv;i++)
        path[i] = Init;
    // 初始化距离
    for(int i=0;i<=Nv;i++)
        dist[i] = Inf;
    // 初始化收录情况
    for(int i=1;i<=Nv;i++)
        collected[i] = false;
    printf("Enter Ne:");
    cin>>Ne;
    // 初始化点
    for(int i=0;i<Ne;i++){
        printf("Enter each edge:");
        cin>>v1>>v2>>w;
        G[v1][v2] = w;  // 有向图
    }
}

// 初始化距离和路径信息
void create(Vertex s){
    dist[s] = 0;
    collected[s] = true;
    for(int i=1;i<=Nv;i++)
        if(G[s][i]){
            dist[i] = G[s][i];
            path[i] = s;
        }
}

// 查找未收录顶点中dist最小者
Vertex FindMin(Vertex s){
    int min = 0;  // 之前特地把 dist[0] 初始化为正无穷
    for(Vertex i=1;i<=Nv;i++)
        if(i != s && dist[i] < dist[min] && !collected[i])
            min = i;
    return min;
}


void Dijkstra(Vertex s){
    create(s);
    while(true){
        Vertex V = FindMin(s);   // 找到
        if(!V)
            break;
        collected[V] = true;  //收录
        for(Vertex W=1;W<=Nv;W++)
            if(!collected[W] && G[V][W]){  // 如果未被收录
                if(dist[V] + G[V][W] < dist[W]){
                    dist[W] = G[V][W] + dist[V];
                    path[W] = V;
                }
            }
    }
}

void output(){
    for(int i=1;i<=Nv;i++)
        cout<<dist[i]<<" ";
    cout<<endl;
    for(int i=1;i<=Nv;i++)
        cout<<path[i]<<" ";
    cout<<endl;
}


int main(){
    build();
    Dijkstra(1);
    output();
    return 0;
}

效果