Acwing 850. Dijkstra求最短路 II

Acwing 850. Dijkstra求最短路 II

链接:850. Dijkstra求最短路 II - AcWing题库

/*
题解:堆优化版本的dijkstra
就是优化了 每次寻找当前距离源最近的点的时间 有小顶堆来维护当前为确定的点和源的距离
那么每次找最新确定的点的时候就是堆顶 
*/
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;
const int N=5e5;
const int inf= 0x3f3f3f3f;
typedef pair<int ,int > PII;
int h[N],ne[N],e[N],w[N],idx;
int n,m;
int st[N];
int dist[N];
void add(int a,int b,int c){
ne[idx]=h[a];
e[idx]= b;
w[idx]= c;
h[a]= idx++;
}
void dijkstra(){
	memset(st,0,sizeof st);
	memset(dist,0x3f,sizeof	dist);
dist[1]=0;
	priority_queue<PII,vector<PII>,greater<PII>> heap;
	heap.push({0,1});
	while(heap.size()){
PII t= heap.top();
int ver= t.second;
int distance= t.first;
		heap.pop();
		if(st[ver])continue;
st[ver]=1;
for(int i=h[ver];i!=-1;i=ne[i]){//m次
int j= e[i];
			if(dist[j]>distance+w[i]){
dist[j]= distance+w[i];
				heap.push({dist[j],j});
			}
		}
	}
if(dist[n]== inf){
		cout<<-1<<endl;
	}else cout<<dist[n]<<endl;
}
int main()
{
    cin>>n>>m;
    memset(h,-1,sizeof h);
    while(m--){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    dijkstra();
    return 0;
}