850. Dijkstra求最短路 II

Posted 2021-05-14 幽殇默

堆优化版dijkstra

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<map> 
#include<queue>
using namespace std;
typedef pair<int,int> PII; 
const int N=1e5+60000;
const int M=1e5+60000;
int n,m;
int h[N],e[M],ne[M],w[M],idx;
int dist[N];
bool st[N];
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 distance=t.first,vis=t.second;
		if(st[vis]) continue;
		st[vis]=true;
		for(int i=h[vis];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>distance+w[i]) //注意这里的 w[i] 不要写成 w[j]
			{
				dist[j]=distance+w[i];
				heap.push({dist[j],j});
			}
		}
	}
	if(dist[n]==0x3f3f3f3f) return -1;
	else return dist[n];
}
int main(void)
{
	cin>>n>>m;
	memset(h,-1,sizeof h);
	while(m--) 
	{
		int a,b,c; cin>>a>>b>>c;
		add(a,b,c);
	}
	cout<<Dijkstra()<<endl;
	return 0;
}

其实你会发现: 堆优化版的Dijkstra算法 它用的是邻接表存的，它的更新其实就是这个点周围点的更新。
朴素Dijkstra其实本质也是这样的。