【51nod 2673】【dijkstra】【堆优化】最短路径

51nod 2673 最短路径

---

题目

给出一张包含n个节点、m条边的无向图，请你求出图上两点s,t间的最短路径长度。

(请大家自行处理重边和自环)

输入
第一行两个数n,m，分别表示节点数和边数，以空格隔开，其中1≤n≤500,1≤m≤50000；
之后m行，每行3个数u,v,w[i]，表示点u和v间有一条权值为w[i]的边，
其中1≤u,v≤n，1≤w[i]≤500000；
最后一行，两个数s,t表示选择的两个点，以空格隔开。
输出
输出一个数，表示s,t间最短路径的长度。
数据范围
对于27%的数据，n≤10，m≤50；
对于70%的数据，n≤100，m≤5000；
对于100%的数据，n≤500，m≤50000，1≤w[i]≤500000；
输入样例

4 3
1 2 6
1 3 4
2 4 2
3 4

输出样例

12

---

解题思路

套用最短路模板应该就可以了
这里用了Dijkstra + 堆优化

---

Code

#include <bits/stdc++.h>
#define N 100000 

using namespace std;

struct DT{
	int to, z, next;
}a[N + 200];
int n, m, x, y, num, S, T, head[N + 200];
int z, dis[N + 200];
priority_queue<pair<int, int> > q;  //大根堆，取绝对值变成小根堆

void add(int x, int y, int z) {
	a[++ num] = (DT){ y, z, head[x] };
	head[x] = num;
}

void dij(int x) {
	memset(dis, 0x3f, sizeof(dis));
	dis[x] = 0;
	q.push(make_pair(0, x));
	while(!q.empty()) {
		int d = -q.top().first, x = q.top().second;
		q.pop();
		if(dis[x] != d) continue;  //表示这个松弛已经不够优了
		for(int i = head[x]; i; i = a[i].next)
			if(dis[a[i].to] > d + a[i].z) {
				dis[a[i].to] = d + a[i].z;
				q.push(make_pair(-dis[a[i].to], a[i].to));  //可以直接维护出最小的松弛
			}
	}
}

int main() {
	scanf("%d %d", &n, &m);
	for(int i = 1; i <= m; i ++) {
		scanf("%d %d %d", &x, &y, &z);
		add(x, y, z), add(y, x, z);
	}
	scanf("%d", &S);
	dij(S);
	scanf("%d", &T);
	printf("%d", dis[T]);
}