Dijkstra算法

问题描述

Description
给定n个点，m条有向边
求每个点到1号点的最短距离
Input
第一行两个数为n,m，n表示顶点个数，m表示边的条数。 (1 ≤ n, m ≤ 100 )
接下来m行，每一行有三个数t1、t2 和t3，表示顶点t1到顶点t2的路程是t3。请注意这些t1->t2是单向的。
Output
输出N个数，分别为每个点到1号点的距离
Sample Input
6 9
1 2 1
1 3 12
2 3 9
2 4 3
3 5 5
4 3 4
4 5 13
4 6 15
5 6 4
Sample Output
0 1 8 4 13 17

算法思路

算法的基本思想是：每次找到离源点（上面例子的源点就是1号顶点）最近的一个顶点，然后以该顶点为中心进行扩展，最终得到源点到其余所有点的最短路径。基本步骤如下：

• 1 将所有的顶点分为两部分：已知最短路程的顶点集合P和未知最短路径的顶点集合Q。最开始，已知最短路径的顶点集合P中只有源点一个顶点。我们这里用一个book[ i ]数组来记录哪些点在集合P中。例如对于某个顶点i，如果book[ i ]为1则表示这个顶点在集合P中，如果book[ i ]为0则表示这个顶点在集合Q中。
• 2 设置源点s到自己的最短路径为0即dis=0。若存在源点有能直接到达的顶点i，则把dis[ i ]设为e[s ][ i ]。同时把所有其它（源点不能直接到达的）顶点的最短路径为设为∞。
• 3在集合Q的所有顶点中选择一个离源点s最近的顶点u（即dis[u ]最小）加入到集合P。并考察所有以点u为起点的边，对每一条边进行松弛操作。例如存在一条从u到v的边，那么可以通过将边u->v添加到尾部来拓展一条从s到v的路径，这条路径的长度是dis[u ]+e[u ][v ]。如果这个值比目前已知的dis[v]的值要小，我们可以用新值来替代当前dis[v ]中的值。
• 4 重复第3步，如果集合Q为空，算法结束。最终dis数组中的值就是源点到所有顶点的最短路径。

一句话来讲就是松弛最近点的所有的出边

算法实现

#include <iostream>
#include <climits>
using namespace std;

int inf = INT_MAX;

int main()
{
    int datas[105][105];
    int dis[10000] = {inf};
    int book[10000] = {0};
    int n,m;
    int t1,t2,t3;
    int minn;
    int mind;

    // n = 6,m = 9;
    cin >> n >> m;

    // 初始化矩阵
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(i==j){
                datas[i][j] = 0;
            }else{
                datas[i][j] = inf;
            }
        }

    }
    // 输入边数
    for(int i=0;i<m;i++){
        cin >> t1 >> t2 >> t3;
        datas[t1-1][t2-1] = t3;
    }
    // 估计距离
    for(int i=0;i<n;i++){
        dis[i] = datas[0][i];
        // cout << dis[i] << "  ";
    }
    book[0] = 1;
    for(int k=0;k<n-1;k++){
        // 找到最近的点
        minn = inf;
        for(int i=1;i<n;i++){
            if(minn > dis[i] && book[i]==0){
                minn = dis[i];
                mind = i;
            }
        }
        // !!!必须找到最短的边之后才对其进行标记
        book[mind] = 1;
        // 对所有的出边进行松弛
        for(int j=0;j<n;j++){
            if(datas[mind][j] < inf)
                if(dis[j] > dis[mind] + datas[mind][j]){
                    dis[j] = dis[mind] + datas[mind][j];
                }
        }
    }
    // 输出最后的结果
    for(int i=0;i<n;i++){
        cout << dis[i] << " ";
    }
    return 0;
}