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

9 篇文章 0 订阅

【模板】单源最短路径(标准版)

题目背景

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

然后呢?

100 → 60 100 \rightarrow 60 10060

Ag → Cu \text{Ag} \rightarrow \text{Cu} AgCu

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

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

题目描述

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

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

输入格式

第一行为三个正整数 n , m , s n, m, s n,m,s
第二行起 m m m 行,每行三个非负整数 u i , v i , w i u_i, v_i, w_i ui,vi,wi,表示从 u i u_i ui v i v_i vi 有一条权值为 w i w_i wi 的有向边。

输出格式

输出一行 n n n 个空格分隔的非负整数,表示 s s 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 ≤ n ≤ 1 0 5 1 \leq n \leq 10^5 1n105

1 ≤ m ≤ 2 × 1 0 5 1 \leq m \leq 2\times 10^5 1m2×105

s = 1 s = 1 s=1

1 ≤ u i , v i ≤ n 1 \leq u_i, v_i\leq n 1ui,vin

0 ≤ w i ≤ 1 0 9 0 \leq w_i \leq 10 ^ 9 0wi109,

0 ≤ ∑ w i ≤ 1 0 9 0 \leq \sum w_i \leq 10 ^ 9 0wi109

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

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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值