迪杰斯特拉算法(Dijkstra)

迪杰斯特拉算法(Dijkstra)

思路：

1. 每次将距离起点v0最近的定点加入集合S，初始时v0位于集合S，体现为标记数组是否已经被标记；
2. 对于尚未求出最短距离的点的集合T，每次从中选取一个距离v0最近的顶点加入S，其路径只能通过S中顶点中转，体现为每选出一个点就更新T中其他的点；

代码实现：

#include <iostream>

#define MAX 1000000
using namespace std;
int arcs[10][10]; // 邻接矩阵
int D[10]; // 最短路径长度
int visited[10]; // 是否求过
int n = 0; // 顶点数
int v0 = 0; // 源点
int v, w; // 中间变量

void dijkstra() {
    // 初始化
    for (v = 0; v < n; ++v) {
        D[v] = arcs[v0][v];
        visited[v] = 0;
    }

    D[v0] = 0;
    visited[v0] = 1; // 将v0放入集合S中
    for (int i= 0; i < n; ++i) {
        // 选点
        int mn = MAX;
        for (w = 0; w < n; ++w) {
            if (!visited[w]) {
                if (D[w] < mn) {
                    mn = D[w];
                    v = w;
                }
            }
        }

        visited[v] = 1; // 将v放入集合S

        for (w = 0; w < n; ++w) {
            if (!visited[w]) {
                if(D[w] > mn + arcs[v][w]) {
                    D[w] = mn + arcs[v][w];
                }
            }
        }

    }
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i++)
    {
         for (int j = 0; j < n; j++)
         {
              cin >> arcs[i][j];
         }
    }
    dijkstra();
    for (int i = 0; i < n; i++) printf("D[%d] = %d\n",i,D[i]);
    return 0;
}