堆是stl中的堆,非手写。
当m和n处于一个量级时,且n的数据量不能够开数组时,可以选择堆优化和邻接表的Dijkstra.
当m是 n 2 n^2 n2时,且 n 2 n^2 n2的大小的数组不会超内存时,可以选择邻接矩阵。
需反复记忆
#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
typedef pair<int,int> pii;
int n,m;
int h[N],e[N],w[N],ne[N],idx;
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 dis[N];
int dijkstra(){
memset(dis,0x3f,sizeof(dis));
dis[1] = 0;
priority_queue<pii, vector<pii>, greater<pii>> heap;//自定义的方式有点不太理解
heap.push({0,1});
//0为距离,1为编号为1的结点。
while(heap.size()){
auto t = heap.top();
heap.pop();
int ver = t.second, dist = t.first;
if(st[ver]) continue;
st[ver] = true;
for(int i = h[ver]; i!=-1; i = ne[i]){//邻接表
int j = e[i];//j为头节点为i的邻居结点
//w[i]为j和i结点之间的距离。
//dist为t这种数据类型中的结点到1结点的距离,加上i到j的距离,来更新1结点到j的距离。
//上述表达,可能有问题
if(dis[j] > dist + w[i]){
dis[j] = dist + w[i];
heap.push({dis[j],j});//更新完后,放入堆中。将最小值弄到最顶部。
}
}
}
if(dis[n] == 0x3f3f3f3f) return -1;
else return dis[n];
}
int main(){
scanf("%d%d",&n,&m);
memset(h,-1,sizeof(h));
while(m--){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
}
int t = dijkstra();
printf("%d\n",t);
return 0;
}