【模板】单源最短路径(标准版)
题目背景
2018 年 7 月 19 日,某位同学在 NOI Day 1 T1 归程 一题里非常熟练地使用了一个广为人知的算法求最短路。
然后呢?
100 → 60 100 \rightarrow 60 100→60;
Ag → Cu \text{Ag} \rightarrow \text{Cu} Ag→Cu;
最终,他因此没能与理想的大学达成契约。
小 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 1≤n≤105;
1 ≤ m ≤ 2 × 1 0 5 1 \leq m \leq 2\times 10^5 1≤m≤2×105;
s = 1 s = 1 s=1;
1 ≤ u i , v i ≤ n 1 \leq u_i, v_i\leq n 1≤ui,vi≤n;
0 ≤ w i ≤ 1 0 9 0 \leq w_i \leq 10 ^ 9 0≤wi≤109,
0 ≤ ∑ w i ≤ 1 0 9 0 \leq \sum w_i \leq 10 ^ 9 0≤∑wi≤109。
本题数据可能会持续更新,但不会重测,望周知。
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;
}