补全：写成小根堆

👁️ 1 人浏览 💬 0 人评论 ❤️ 添加收藏

下面的 Dijkstra 用了不带参数的 priority_queue，那是大根堆，先弹出的是最远的点——在这张小图（0-1:10　0-2:1　1-2:1　1-3:1）上结果就错了。补全小根堆的写法，输出距离表。

（本题用 g++ -std=c++17 -O0 编译。）

提交你的答案

请登录后提交答案。

去登录

← 两个 INF 相加溢出没补全：跳过过期的堆元素 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

#include <algorithm>
#include <iostream>
#include <queue>
#include <set>
#include <string>
#include <tuple>
#include <utility>
#include <vector>
using namespace std;

typedef long long ll;
const ll INF = 1000000000;                       // 走不到的点记这个值，输出时写成 -1
using Graph = vector<vector<pair<int, int>>>;    // g[u] 里放 (邻居 v, 边权 w)

Graph build(int n, const vector<vector<int>>& edges) {
    // 无向图：每条边 {a, b, w} 两个方向都加；邻接表按邻居编号排好
    Graph g(n);
    for (const auto& e : edges) {
        g[e[0]].push_back({e[1], e[2]});
        g[e[1]].push_back({e[0], e[2]});
    }
    for (auto& adj : g) sort(adj.begin(), adj.end());
    return g;
}

string show(const vector<ll>& d) {
    // 距离表拼成 0/2/1/...，走不到的写 -1
    string s;
    for (size_t i = 0; i < d.size(); i++) {
        s += (i ? "/" : "") + (d[i] >= INF ? string("-1") : to_string(d[i]));
    }
    return s;
}

vector<ll> dijkstra(const Graph& g, int s) {
    vector<ll> d(g.size(), INF);
    vector<bool> done(g.size(), false);
    // TODO: 这是大根堆（先出距离最大的），改成小根堆
    priority_queue<pair<ll, int>> pq;
    d[s] = 0;
    pq.push({0, s});
    while (!pq.empty()) {
        auto [du, u] = pq.top();
        pq.pop();
        if (done[u]) continue;
        done[u] = true;
        for (auto [v, w] : g[u]) {
            if (du + w < d[v]) {
                d[v] = du + w;
                pq.push({d[v], v});
            }
        }
    }
    return d;
}

int main() {
    Graph g = build(4, {{0, 1, 10}, {0, 2, 1}, {1, 2, 1}, {1, 3, 1}});
    cout << show(dijkstra(g, 0)) << endl;
}

本次输入：

输出：

👩‍🏫

💬 题目评论

补全：写成小根堆

更多题目

全部评论