堆优化Dijkstra:单源最短路(从一个点到其他所有点的最短路)适用于稀疏图(邻接表)。
步骤:同朴素版Dijkstra,加入堆进行优化
1. 一号点的距离初始化为零,其他点初始化成无穷大。
2. 将一号点放入堆中。
3. 不断循环,直到堆空。每一次循环中执行的操作为:
- 弹出堆顶(与朴素版diijkstra找到S外距离最短的点相同,并标记该点的最短路径已经确定)。
- 用该点更新临界点的距离,若更新成功就更新成功就加入到堆中。
#include <iostream>
#include <queue>
#include <algorithm>
#include <string.h>
using namespace std;
typedef pair<int,int>PII;
const int N=100010;
int n,m;
int h[N],e[N],ne[N],w[N];
int dist[N];
bool mark[N];
int idx;
void add(int a,int b,int c){
e[idx]=b;
w[idx]=c;//添加权重
ne[idx]=h[a];
h[a]=idx++;
}
int dijkstra(){
//初始化距离
memset(dist,0x3f,sizeof(dist));
dist[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;
//点的距离
int distance=t.first;
//如果当前点已经使用过,则为冗余点 则跳过
if(mark[ver]) continue;
//用当前点更新其他点的距离
for(int i=h[ver];i!=-1;i=ne[i]){
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));
scanf("%d%d", &n, &m);
while (m--)
{
int x, y, c;
scanf("%d%d%d", &x, &y, &c);
add(x, y, c);
}
cout << dijkstra() << endl;
return 0;
}