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

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

题目背景

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$。

本题数据可能会持续更新，但不会重测，望周知。

2018.09.04 数据更新 from @zzq

思路： 一看到单源最短路径就能想到dijkstra，于是就有了下面的代码：

#include <bits/stdc++.h>
using namespace std;
int n, m, s, ans[100005];
bool vis[100005];
struct node{
    int u, v;
};
vector<node> vec[100005];
int main(){
    scanf("%d %d %d", &n, &m, &s);
    for(int i = 0; i < m; i++){
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        node d{b, c};
        vec[a].push_back(d);
    }
    for(int i = 1; i <= n; i++){
        ans[i] = 0x7f7f7f7f;
    }
    int cur;
    ans[s] = 0;
    for(int i = 0; i < n; i++){
        int minn = 2147483647;
        for(int j = 1; j <= n; j++){
            if(!vis[j] && ans[j] < minn){
                minn = ans[j];
                cur = j;
            }
        }
        vis[cur] = true;
        for(int j = 0; j < vec[cur].size(); j++){
            ans[vec[cur][j].u] = min(ans[vec[cur][j].u], ans[cur] + vec[cur][j].v);
        }
    }
    for(int i = 1; i <= n; i++){
        printf("%d ", ans[i]);
    }
    return 0;
}

然后提交一看，所有测试点全部超时。

于是堆优化就出现了。由于每次都需要贪心寻找当前没被遍历过的距离起点最近的点，所以可以用优先队列来寻找最小点，寻找最小值的时间复杂度是O(logn)，比直接遍历所有点的O(n)复杂度低。具体来说，就是在更新每个点与起点的距离时，如果更新后的值比原来的值小，就把新值存入优先队列，下一次寻找最小点时直接弹出队列顶部元素即可。

代码：

#include <bits/stdc++.h>
using namespace std;
int n, m, s, ans[100005];
bool vis[100005];
struct node{
    int u, v;
    bool operator<(const node& d) const{
        return v > d.v;
    }
    bool operator>(const node& d) const{
        return v < d.v;
    }
};
vector<node> vec[100005];
priority_queue<node, vector<node> > q;
int main(){
    scanf("%d %d %d", &n, &m, &s);
    for(int i = 0; i < m; i++){
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        node d{b, c};
        vec[a].push_back(d);
    }
    for(int i = 1; i <= n; i++){
        ans[i] = 0x3f3f3f3f;
    }
    ans[s] = 0;
    node d{s, 0};
    q.push(d);
    while(!q.empty()){
        node cur = q.top();
        q.pop();
        if(vis[cur.u]) continue;
        vis[cur.u] = true;
        for(int i = 0; i < vec[cur.u].size(); i++){
            if(ans[vec[cur.u][i].u] > ans[cur.u] + vec[cur.u][i].v){
                ans[vec[cur.u][i].u] = ans[cur.u] + vec[cur.u][i].v;
                node tmp = {vec[cur.u][i].u, ans[vec[cur.u][i].u]};
                q.push(tmp);
            }
        }
    } 
    for(int i = 1; i <= n; i++){
        printf("%d ", ans[i]);
    }
    return 0;
}