单源最短路裸题+堆优化dijkstra模板
堆优化dijkstra:将找最小距离的过程优化,不断取出到源点最小距离的节点,然后用这个节点更新他的临点。在取出时标记st为true
#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int N=15010;
typedef pair<int,int> PII;
int h[N],e[N],ne[N],w[N],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(int start,int end){
memset(dist,0x3f,sizeof dist);
dist[start]=0;
//小根堆
priority_queue<PII,vector<PII>,greater<PII>> heap;
heap.push({0,start});
//要在数据取出后设置访问
//st[start]=true;
while(heap.size()){
auto t=heap.top();
heap.pop();
//这个点到源点的距离
int d=t.first,ver=t.second;
if(st[ver]) continue;
//出堆的时候设置为true
st[ver]=true;
//遍历这个点的所有出边
for(int i=h[ver];~i;i=ne[i]){
int j=e[i];
if(dist[j]>dist[ver]+w[i]){
dist[j]=dist[ver]+w[i];
heap.push({dist[j],j});
}
}
}
return dist[end];
}
int main(){
int n,m,start,end;
cin>>n>>m>>start>>end;
memset(h,-1,sizeof h);
for(int i=0;i<m;i++){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
int t = dijkstra(start,end);
printf("%d\n",t);
return 0;
}