#define _CRT_SECURE_NO_WARNINGS 1
//堆优化版的dijkstra算法
//稀疏图 用邻接表储存
//主要是优化找找出最短一点的路径这一步 用堆来选择
//时间复杂度O(mlogn)
//建堆两种方式,1.手写堆2.优先队列
//在这里用stl优先队列 priority queue
#include<iostream>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1000010;
int n, m;
int e[N], h[N], ne[N], idx, w[N];//w[N]数组是边的权重
int dist[N];
bool st[N];
typedef pair<int, int>PII;//first为距离,second为点
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数组 每个点的距离均为INF
dist[1] = 0;//1并入最短路径 距离为0
priority_queue<PII, vector<PII>, greater<PII>>heap;//定义一个小根堆heap的标准句式,记忆就行 用vector来实现
heap.push({ 0,1 });//优先队列排序是按照pair 的first排序 所以first存路径长
while (heap.size())//堆不空时开始循环
{
auto t = heap.top();//t来接受对顶元素,因为是小顶堆,对顶元素距离一定最小,省去遍历的时间
heap.pop();//出堆
int ver = t.second, distance = t.first;//ver点的下标,distance距离
if (st[ver]) continue;//消除重边的影响
st[ver] = true;
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];//w[i]就是编号为i发出的点的权重
heap.push({ dist[j],j });
}
}
}
if (dist[n] == 0x3f3f3f3f) return -1;
return dist[n];
}
int main()
{
memset(h, -1, sizeof h);
scanf("%d%d", &n, &m);
while (m--)
{
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
add(x, y, z);
}
int d = dijkstra();
cout << d << endl;
return 0;
}
acwing 850. Dijkstra求最短路 II
最新推荐文章于 2024-10-08 14:26:15 发布