Acwing_850（Dijkstra求最短路 II）

原题链接：

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

题解：

本题的数据规模为：

因此可以采用堆优化的Dijkstra算法，需要注意的是采用vector开辟邻接矩阵会Segmentation Fault，因此可以通过手搓单链表建立邻接矩阵

代码：

#pragma GCC optimize(2, 3, "Ofast", "inline")
#include<bits/stdc++.h>
using namespace std;
using PII = pair<int, int>;
const int N = 2e5 + 10;
int h[N], e[N], ne[N], idx, dis[N], v[N], state[N];
//h数组是头结点数组，存储的邻接表众多链表的头结点下标（注意是下标，非编号）
//e数组存储的是下标对应的编号
//ne数组存储的是当前下标对应的下一个下标
//v数组存储的是当前下标对应的边权
int n, m;

void add(int a, int b, int c) {
	e[idx] = b, v[idx] = c, ne[idx] = h[a], h[a] = idx++;
}

int dijkstra2() {
	memset(dis, 0x3f, sizeof(dis));
	dis[1] = 0;

	priority_queue<PII, vector<PII>, greater<PII>> heap;
	heap.push({ 0,1 });//距离-编号

	while (heap.size()) {
		auto F = heap.top();
		heap.pop();

		int var = F.second;

		if (state[var]) continue;
		state[var] = 1;

		for (int i = h[var];i != -1;i = ne[i]) {
			int cur = e[i];
			if (dis[cur] > dis[var] + v[i]) {
				dis[cur] = dis[var] + v[i];
				heap.push({ dis[cur],cur });
			}
		}
	}
	if (dis[n] == 0x3f3f3f3f)
		return -1;
	else
		return dis[n];
}


int main() {
	cin >> n >> m;
	memset(h, -1, sizeof(h));
	for (int i = 0;i < m;i++) {
		int x, y, z;cin >> x >> y >> z;
		add(x, y, z);
	}
	cout << dijkstra2();
}