堆优化dijkstra(稀疏图)

题目:
在这里插入图片描述
时间复杂度: mlogn, 一共有m条边,在堆中修改某个值logn

Q: 为啥使用堆?
A: 在堆中找到最大最小值是O(1)的, 由于朴素版瓶颈在查找所有点到当前点距离最近的点,因此利用堆就可以从O(n2)优化到O(1), 但找到后要修改这个点的所有出边是否能更新到源点的距离,修改一次就是logn

Q: 为什么使用PII?
A: 最终目的是最后一个节点到源点的距离,利用堆就是为了快速查找最小的距离,所以肯定要存距离,为了标识到源点的距离和找到某个点的所有出边,所以要存节点编号。

Q: 为什么堆中会存放冗余元素,如何处理?
A: 某一点确定后就会更新它的邻边的点到源点的距离(初始化刚开始d[]是INF),每次更新后还会把更新过的点放入堆中。继续取出堆中的元素发现,取出元素的邻边,发现能更新(但并不是由于INF的更新,而是通过该点作为桥梁来更新),因此会把之前加入过的点再次加入(节点编号相同,距离不同),产生冗余。由于堆自动实现了排序功能,所以尽管节点编号是相同的,但取出的是最小的距离,因此在更新完某个点的所有出边后,将该点打上标记(已经找到了该点到源点的最小距离),所以在下一次取出堆中元素时判断该元素是否被上标记就ok了,如果打上过标记就没必要再次更新其邻边了

Q: 只更新了一次某个点到邻边的距离,那如果该点到源点的距离再次被更新了,会对后面更新过的点会产生影响么?
A: 不会,堆每次取出的都是最小值,即使原来的点被更新后,从堆中拿出来的也是新的(更新过的)点

代码:

#include <iostream>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
const int N = 150010;
typedef pair<int, int> PII;
int h[N], e[N], ne[N], w[N], idx;
int d[N];
bool st[N];
int n, m;


void add(int a, int b, int c)
{
    w[idx] = c;
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++;
}

int dij()
{
    memset(d, 0x3f, sizeof d);
    d[1] = 0;
    priority_queue<PII, vector<PII>, greater<PII>> heap;	//定义小根堆
    heap.push({0, 1});
    
    while(heap.size())
    {
        auto t = heap.top();
        heap.pop();
        int dis = t.first, ver = t.second;
        if(st[ver]) continue;	//判断该点的邻边是否被更新过
        
        st[ver] = true;		//打上标记,表示该点的邻边已经更新过了
        for(int i = h[ver]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(d[j] > dis + w[i])   
            {
                d[j] = dis + w[i];
                heap.push({d[j], j});
            }
        }
    }
    if(d[n] == 0x3f3f3f3f)  return -1;
    else return d[n];
}

int main()
{
    cin >> n >> m;
    memset(h, -1, sizeof h);
    while(m -- )
    {
        int a, b, w;
        cin >> a >> b >> w;
        add(a, b, w);
    }
    
    cout << dij() << endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值