图论单源最短路径——堆优化版迪杰斯特拉算法（dijkstra）

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

题目背景

2018 年 7 月 19 日，某位同学在 NOI Day 1 T1 归程 一题里非常熟练地使用了一个广为人知的算法求最短路。

然后呢？

$100\to 60$；

$\text{Ag}\to \text{Cu}$；

最终，他因此没能与理想的大学达成契约。

小 F 衷心祝愿大家不再重蹈覆辙。

题目描述

给定一个 $n$ 个点，$m$ 条有向边的带非负权图，请你计算从 $s$ 出发，到每个点的距离。

数据保证你能从 $s$ 出发到任意点。

输入格式

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

输出格式

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

样例 #1

样例输入 #1

4 6 1
1 2 2
2 3 2
2 4 1
1 3 5
3 4 3
1 4 4

样例输出 #1

0 2 4 3

提示

样例解释请参考 数据随机的模板题。

$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 2024.4.29 SimonAN
*/
#include<iostream>
#include<bits/stdc++.h>
#define lyh(i,a,b) for(int i = a;i <= b;i ++)
#define hyl(i,a,b) for(int i = a;i >= b;i --)
#define debug(a) cout<<#a<<'='<<a<<endl;
#define endl "\n"
#define LL long long
#define INF 0x3f
using namespace std;
const int N = 5e5 + 100;
int n,m,s;
int h[N],e[N],ne[N],idx;
LL w[N];
int d[N];
void add(int a,int b,int c){
	e[idx] = b, w[idx] = c,ne[idx] = h[a], h[a] = idx ++;
}
int q[N];
bool st[N];
int tt = -1,hh = 0;
typedef pair<int,int> PII;
void dijkstra() {
    memset(d, 0x3f, sizeof d);
    d[1] = 0;
    // 定义小根堆
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    heap.push({0, 1}); // 先将最短的距离添加到小根堆中
    
    while (heap.size()) {
        auto t = heap.top(); // 取出当前已经确定的最短距离的点
        heap.pop();
        
        int ver = t.second;
        if (st[ver]) continue; // 如果已经被标记过，说明当前的点的边是比较长的重边，直接跳过。
        st[ver] = true;
        
        // 使用当前距离最短的点来更新其出边
        for (int i = h[ver]; i != -1; i = ne[i]) {
            int j = e[i]; // j 为 ver 出边所到达的点
            if (d[j] > d[ver] + w[i]) {
                /* 如果源点直接到 j 点的距离比源点先到 ver 点再从 ver 点到 j 点的距离大，
                    那么就更新 dist[j]，使dist[j] 到源点的距离最短
                */
                d[j] = d[ver] + w[i];
                // 将下一个距离源点最近的点添加到小根堆中
            }
            heap.push({d[j], j});//这一步写在外面 
        }
    }
}

int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	memset(d,0x3f3f3f,sizeof d);
	memset(h,-1,sizeof h);
	cin>>n>>m>>s;
	lyh(i,1,m){
		int a,b,c; cin>>a>>b>>c;
		add(a,b,c);
	}
	dijkstra();
	lyh(i,1,n){
//		
//		
//		if(d[i] >= 0x3f3f3f) cout<<INT_MAX<<" ";//INT_MAX 是最大的int值也就是2^31-1
//		else cout<<d[i]<<" ";	
		cout<<d[i]<<" ";
	}
}