题目:
时间复杂度: 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;
}