【洛谷】P4779 【模板】单源最短路径（标准版）

题目地址：

https://www.luogu.com.cn/problem/P4779

题目描述：
给定一个$n$个点，$m$条有向边的带非负权图，请你计算从$s$出发，到每个点的距离。数据保证你能从$s$出发到任意点。

输入格式：
第一行为三个正整数$n,m,s$。 第二行起$m$行，每行三个非负整数$u_i,v_i$，表示从$u_i$到$v_i$有一条权值为$w_i$的有向边。

输出格式：
输出一行$n$个空格分隔的非负整数，表示$s$到每个点的距离。

数据范围：
$1\le n\le 10^5$
$1\le m\le 2\times 10^5$
$s=1$
$1\le u_i,v_i\le n$
$0\le w_i\le 10^9$
$0\le \sum w_i\le 10^9$

由题目数据知图是稀疏图，要用堆优化Dijkstra算法来做。代码如下：

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

typedef pair<int, int> PII;

const int N = 1e5 + 10, M = 2e5 + 10;
int n, m, S;
int h[N], e[M], ne[M], w[M], idx;
int dist[N];
bool st[N];

void add(int a, int b, int c) {
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

void dijkstra(int S) {
    priority_queue<PII, vector<PII>, greater<PII> > heap;
    memset(dist, 0x3f, sizeof dist);
    dist[S] = 0;

    heap.push({0, S});
    while (heap.size()) {
        auto t = heap.top(); heap.pop();
        int v = t.second, d = t.first;
        if (st[v]) continue;

        st[v] = true;
        for (int i = h[v]; ~i; i = ne[i]) {
            int j = e[i];
            if (!st[j] && dist[j] > d + w[i]) {
                dist[j] = d + w[i];
                heap.push({dist[j], j});
            }
        }
    }
}

int main() {
    cin >> n >> m >> S;

    memset(h, -1, sizeof h);
    for (int i = 0; i < m; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
    }

    dijkstra(S);

    for (int i = 1; i <= n; i++) cout << dist[i] << ' ';
    cout << endl;

    return 0;
}

时间复杂度$O(m\log n)$，空间$O(n)$。