优先队列优化 Dijkstra
给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环,所有边权均为非负值。
请你求出 1 号点到 n 号点的最短距离,如果无法从 1 号点走到 n 号点,则输出 −1。
输入格式
第一行包含整数 n 和 m。
接下来 m 行每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。
输出格式
输出一个整数,表示 1 号点到 n 号点的最短距离。
如果路径不存在,则输出 −1。
数据范围
1≤n,m≤1.5×10^5,
图中涉及边长均不小于 0,且不超过 10000。
数据保证:如果最短路存在,则最短路的长度不超过 109。
输入样例:
3 3
1 2 2
2 3 1
1 3 4
输出样例:
3
堆优化版dijkstra适合稀疏图
思路
堆优化版的dijkstra是对朴素版dijkstra进行了优化,在朴素版dijkstra中时间复杂度最高的寻找距离
最短的点O(n^2)可以使用最小堆优化。
- 一号点的距离初始化为零,其他点初始化成无穷大。
- 将一号点放入堆中。
- 不断循环,直到堆空。每一次循环中执行的操作为:
弹出堆顶(与朴素版diijkstra找到S外距离最短的点相同,并标记该点的最短路径已经确定)。
用该点更新临界点的距离,若更新成功就加入到堆中。
时间复杂度分析
寻找路径最短的点:O(n)O(n)
加入集合S:O(n)O(n)
更新距离:O(mlogn)
#include<bits/stdc++.h>
using namespace std;
typedef pair<int ,int > PII;
const int N = 150010;
//用领接表存储稀疏图
int h[N], e[N], w[N], ne[N], idx;
int dist[N];
bool st[N];
int n, m;
void add(int x, int y, int z){
e[idx] = y;
w[idx] = z;
ne[idx] = h[x];
h[x] = idx ++;
}
int Dijkstra(){
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
priority_queue<PII, vector<PII>, greater<PII>> heap;
//0是权重,1是idx
heap.push({0, 1});
while(heap.size()){
PII k = heap.top();
heap.pop();
int ver= k.second, distance = k.first;
if(st[ver]) continue;
st[ver] =true;
for(int i = h[ver]; i != -1; i = ne[i]){
//j表示 ver的下一个是j
int j = e[i];
if(dist[j] > distance + w[i]){
dist[j] = distance + w[i];
heap.push({dist[j], j});
}
}
}
if(dist[n] == 0x3f3f3f3f) return -1;
else return dist[n];
}
int main(){
memset(h, -1, sizeof h);
cin >> n >> m;
while(m -- ){
int x, y, z;
cin >> x >> y >> z;
add (x, y, z);
}
cout << Dijkstra() << endl;
return 0;
}