【模版】堆优化Dijkstra求最短路

【模版】堆优化Dijkstra求最短路

关于堆优化

一：堆优化的主要思想就是使用一个优先队列（就是每次弹出的元素一定是整个队列中最小的元素）来代替最近距离的查找，用邻接表代替邻接矩阵，这样可以大幅度节约时间开销。
几个细节：
优先队列的数据类型
优先队列应该用于快速寻找距离最近的点。由于优先队列只是将最小的那个元素排在前面，因此应该定义一种数据类型，使得它包含该节点的编号以及该节点当前与起点的距离。

在什么时候对队列进行操作
队列操作的地方，首先就是搜索刚开始，要为起点赋初始值，此时必须将起点加入优先队列中。该队列元素的节点编号为起点的编号，该节点当前与起点的距离为0
。

如果一个节点到起点的最短距离通过其他的运算流程发生了变化，如何处理队列中的那个已经存入的元素
事实上，不需要理会队列中的元素，而是再存入一个就行了。因为如果要发生变化，只能将节点与起点之间的距离变得更小，而优先队列恰好是先让最小的那个弹出。
因此，轮到某一个队列元素弹出的时候，如果有多个元素的节点编号相同，那么被弹出的一定是节点编号最小的一个。等到后面再遇到这个节点编号的时候，我们只需要将它忽略掉就行了。
来源：https://blog.csdn.net/rentenglong2012/article/details/78463723

二：采用优先队列去优化。意思就是当你在A的时候，发现AB=3.5，AF=6，另外的1000，都更新他们，再把他们都丢进优先队列，而优先队列会自动替你把这一串，由小到大排好序。所以你只需要取队列的队首元素就可以了，省去了你找到最小值的过程。我们使用优先队列就是节省了排序的时间。此外你可以用Vector或者邻接表，不用像二维数组一样开一个矩阵，会更快。
来源：https://blog.csdn.net/qq_36523667/article/details/78779272

模版参考

https://blog.csdn.net/Mashiro_ylb/article/details/78287724

测试题目

P4779 【模板】单源最短路径（标准版）

代码

//---------使用模版前言---------
//注意顶点数MAXN的值 注意极大数据inf的值
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <utility>
#include <algorithm>
#include <vector>
#define MAXN 300010     //最大顶点数 注意依题修改大小
#define inf 0x3f3f3f3f   //极大数据 10^9数量级
using namespace std;
typedef pair<int, int> info;    //存放当前点及从源点到当前点的最短路径的信息
priority_queue<info, vector<info>, greater<info> > Q;
//优先队列实现将已确定最短路径的点放入队列 小顶堆实现堆优化
struct edge{    //定义弧信息
    int to;     //弧指向的点
    int weight;     //弧的权重
    edge(int t, int w): to(t), weight(w){}  //弧的初始化
};
vector<edge> G[MAXN];   //存放当前点出发的每一条弧的信息
int n, m, s;    //点 弧 源点
int dis[MAXN];  //从源点到当前点的最短路径(迪杰斯特拉跑完之前不一定是最终解)
int vis[MAXN];  //该数组用于标记当前点是否已访问过 起到优化作用
void Dijkstra(int s){   //s为源点，使用Dijkstra查找源点到其他任意点的最短路径
    memset(vis, 0, sizeof(vis));
    //每跑一次算法重置访问标记数组 跑一次算法可得出一个源点到其他任意点的最短路径
    for (int i = 1; i <= n; i++) dis[i] = inf;  //初始化源点到任意其他点最短路径无穷大
    dis[s] = 0; //源点到源点最短路径为0
    Q.push(make_pair(0, s));    //将源点放入队列
    while (!Q.empty()){ //结束条件为队列里无未确定最短路径点
        info u = Q.top();
        Q.pop();    //取出并去除队头元素
        int x = u.second;   //当前点编号
        if(vis[x]) continue;    //如果当前点已访问则忽略该点
        vis[x] = 1; //标记当前点已访问
        for (int i = 0; i < G[x].size(); i++){  //遍历从x点出发的所有弧
            edge e = G[x][i];   //从x点出发的第i条弧
            if (dis[e.to] > dis[x] + e.weight){
                //若经过x点的松弛使得从源点到弧尾指向的点的最短路径更短 则更新最短路径
                dis[e.to] = dis[x] + e.weight;
                Q.push(make_pair(dis[e.to], e.to)); //将已确定最短路径的点放入队列
            }
        }
    }
}
int main(){
    scanf("%d%d%d", &n, &m, &s);    //点 弧 源点
    int from, to, weight;   //弧的起点 终点 权重
    for (int i = 1; i <= m; i++){
        scanf("%d%d%d", &from, &to, &weight);
        edge in(to, weight);
        G[from].push_back(in);  //往图中加入所有已知弧的信息
    }
    Dijkstra(s);    //算法实施
    for (int i = 1; i < n; i++) printf("%d ", dis[i]);
    printf("%d\n", dis[n]);     //规范输出无多余空格
    //输出从源点到其他所有点的最短路径
    return 0;
}