AcWing850 dijkstra求最短路2

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef pair<int, int> PII;
const int N = 1e6 + 10;
int h[N], e[N], ne[N], idx;
int w[N];//存储权重
bool st[N];//标记这个点的最短路是否已经被找到
int dist[N];
int n, m;
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中为什么要存pair呢，首先小根堆是根据距离来排的，所以有一个变量要是距离，其次在从堆中拿出来的时
	// 候要知道知道这个点是哪个点，不然怎么更新邻接点呢？所以第二个变量要存点。
	heap.push({ 0, 1 }); // 这个顺序不能倒，pair排序时是先根据first，再根据second，这里显然要根据距离排序
	while (heap.size()) {
		PII k = heap.top();
		heap.pop();
		int ver = k.second, distance = k.first;

		if (st[ver])continue;//如果当前点的最短路已经确定那就continue；
		//如果不这一步的话可能会有某个点既是a的邻接点又是b的邻接点会在下面的for循环中多次加入
		//但是由于每次的判断 总是取的是最小 会tle但不会答案错误
		st[ver] = true;
		for (int i = h[ver]; i != -1; i = ne[i]) {
			int j = e[i];
			if (dist[j] > distance + w[i])
			{
				dist[j] = distance + w[i];
				heap.push({ dist[j], j });
			}
		}
	
	}
	if (dist[n] == 0x3f3f3f3f) return -1;
		else return dist[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);
	}

	cout << dijkstra() << endl;

	return 0;
}