堆优化版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其实本质也是这样的。