算法学习整理(一) Dijkstra

假设构建的是一副无向图

图中有0~6 7个点，每段线表示从一个点到另一个点的值
Dijkstra算法的目的是找到一个点到另一个点的最短路径

假设已经构建好了图的关系，并选取了起始点a1
思路：

1. 初始化两个点集合S和U，S包括起点a1，U包括剩余点，遍历计算S中点到U中点的最短距离，不相邻/连接的距离为∞；
2. 上一步计算得到U集中所有点的距离，从U中选出最短距离点k，从U集中移动到S集中；
3. 重复上述1和2的步骤，至U为空。

我们假设输入为 N 个点，并输入m个关系，先假设每个关系之间无向，然后连续输入m个点的关系。最后输入一个点，求出到其他点的具体距离。
输入样例

7 12
0 1 12
0 5 16
0 6 14
1 2 10
1 5 7
2 3 3
2 4 5
2 5 6
3 4 4
4 5 2
4 6 8
5 6 1000
0

#include <iostream>
#include <vector>
#include <deque>
using namespace std;

int main()
{
    int N, m;
    while(cin >> N >> m)
    {
        vector< vector<int> > map(N, vector<int>(N, -1));
        int s;
        for(int i = 0; i < m; i++)
        {
            int a, b, d;
            cin >> a >> b >> d;
            map[a][b] = d;
            map[b][a] = d;//<这里是无向图，有向图注释掉
        }
        cin >> s;
        vector<int> dis(N, 1000000);
        deque<int> u;
        deque<int> end;
        end.push_back(s);
        dis[s] = 0;
        for(int i = 0; i < N; i++){
            if(i == s) continue;
            u.push_back(i);
        }
        while(u.size() > 0)
        {
            for(int i = 0; i < end.size(); i++)
            {
                for(int j = 0; j < u.size(); j++)
                {
                    {
                        if(map[end[i]][u[j]] > 0)
                        {
                            dis[u[j]] = dis[u[j]] > (map[end[i]][u[j]] + dis[end[i]]) ? (map[end[i]][u[j]] + dis[end[i]]) : dis[u[j]];
                        }
                    }
                }
            }
            int min_dis = 1000000;
            int index = 0;
            // 找最小的下标
            for(int i = 0; i < u.size(); i++)
            {
                if(dis[u[i]] < min_dis)
                {
                    index = i;
                    min_dis = dis[u[i]];
                }
            }
            end.push_back(u[index]);
            u.erase(u.begin() + index);
        }
        for(int i = 0; i < dis.size(); i++)
        {
            cout << "由点 " << s << "到点 " << i << " 的距离 : " << dis[i] << endl;
        }
    }
    return 0;
}

输出