算法训练营第七十二天 | dijkstra（朴素版）、dijkstra（堆优化版）精讲

dijkstra（朴素版）

题目链接：
https://kamacoder.com/problempage.php?pid=1047

dijkstra三部曲：

第一步，选源点到哪个节点近且该节点未被访问过
第二步，该最近节点被标记访问过
第三步，更新非访问节点到源点的距离（即更新minDist数组）

代码如下：

#include <iostream>
#include <vector>
#include <climits>
using namespace std;
 
const int max = INT_MAX;
 
int main() {
    int n, m;
    cin >> n >> m;
    int s, e, v;
    vector<vector<int>> dis(n+1, vector<int>(n+1, INT_MAX));
    vector<int> visited(n+1, 0);
    for (int i = 0; i < m; i++) {
        cin >> s >> e >> v;
        dis[s][e] = v;
    }
    int minDis = INT_MAX;
    int minIdx = 0;
    visited[1] = 1;
    for (int i = 0; i < n-1; i++) {
        minDis = INT_MAX;
        for (int j = 2; j < n+1; j++) {
            if (minDis > dis[1][j] && visited[j] == 0) {
                minDis = dis[1][j];
                minIdx = j;
            }
        }
        visited[minIdx] = 1;
        for (int j = 2; j < n+1; j++) {
            if (j == minIdx) continue;
            if (dis[minIdx][j] != INT_MAX && dis[1][j] > minDis + dis[minIdx][j]) {
                dis[1][j] = minDis + dis[minIdx][j];
            }
        }
    }
    if (dis[1][n] != INT_MAX)
        cout << dis[1][n] << endl;
    else cout << -1 << endl;
}

dijkstra（堆优化版）精讲

题目链接如下：
https://kamacoder.com/problempage.php?pid=1047

堆优化细节

其实思路依然是 dijkstra 三部曲：

第一步，选源点到哪个节点近且该节点未被访问过
第二步，该最近节点被标记访问过
第三步，更新非访问节点到源点的距离（即更新minDist数组）
只不过之前是 通过遍历节点来遍历边，通过两层for循环来寻找距离源点最近节点。 这次我们直接遍历边，且通过堆来对边进行排序，达到直接选择距离源点最近节点。

代码如下：

#include <iostream>
#include <vector>
#include <climits>
#include <queue>
#include <list>
using namespace std;
 
const int max = INT_MAX;
 
struct Edge {
    int to;  
    int val; 
 
    Edge(int t, int w): to(t), val(w) {}  
};
 
class mycomparison {
public:
    bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
        return lhs.second > rhs.second;
    }
};
 
priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pq;
 
int main() {
    int n, m;
    cin >> n >> m;
    int s, e, v;
    vector<list<Edge>> grid(n + 1);
    vector<int> visited(n+1, 0);
    std::vector<int> minDist(n + 1, INT_MAX);
    for (int i = 0; i < m; i++) {
        cin >> s >> e >> v;
        grid[s].push_back(Edge(e,v));
    }
    pq.push(pair<int, int>(1, 0));
    minDist[1] = 0;
    while (!pq.empty()) {
        pair<int, int> cur = pq.top(); pq.pop();
        visited[cur.first] = 1;
        for (Edge edge : grid[cur.first]) { 
            if (!visited[edge.to] && minDist[cur.first] + edge.val < minDist[edge.to]) { // 更新minDist
                minDist[edge.to] = minDist[cur.first] + edge.val;
                pq.push(pair<int, int>(edge.to, minDist[edge.to]));
            }
        }
    }
    if (minDist[n] != INT_MAX)
        cout << minDist[n] << endl;
    else cout << -1 << endl;
}