67 lines
1.9 KiB
C++
67 lines
1.9 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
|
|||
|
|
using namespace std;
|
|||
|
|
/*
|
|||
|
|
(spfa) O(m)
|
|||
|
|
平均O(m)，最坏O(nm) 一般会被卡到O(n*m)会被卡死
|
|||
|
|
*/
|
|||
|
|
const int N = 2510; // 端点数量
|
|||
|
|
const int M = 6200 * 2 + 10; // 边的数量，记得别开太小，和N一样就等着挂吧
|
|||
|
|
int n; // n个城镇
|
|||
|
|
int m; // m条路径
|
|||
|
|
int S; // 起点
|
|||
|
|
int T; // 终点
|
|||
|
|
int h[N], e[M], w[M], ne[M], idx; // 邻接表
|
|||
|
|
int d[N]; // 最短距离数组
|
|||
|
|
queue<int> q; // 队列
|
|||
|
|
bool st[N]; // 是否使用过
|
|||
|
|
|
|||
|
|
// 邻接表模板
|
|||
|
|
void add(int a, int b, int c) {
|
|||
|
|
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// spfa模板
|
|||
|
|
void spfa() {
|
|||
|
|
// 将所有距离初始化为无穷大
|
|||
|
|
memset(d, 0x3f, sizeof d);
|
|||
|
|
// 出发点的距离清零
|
|||
|
|
d[S] = 0;
|
|||
|
|
|
|||
|
|
q.push(S); // 出发点入队列
|
|||
|
|
st[S] = true; // 出发点标识已使用
|
|||
|
|
while (q.size()) {
|
|||
|
|
int u = q.front();
|
|||
|
|
q.pop();
|
|||
|
|
st[u] = false;
|
|||
|
|
for (int i = h[u]; ~i; i = ne[i]) {
|
|||
|
|
int v = e[i];
|
|||
|
|
if (d[v] > d[u] + w[i]) {
|
|||
|
|
d[v] = d[u] + w[i];
|
|||
|
|
if (!st[v]) {
|
|||
|
|
st[v] = true;
|
|||
|
|
q.push(v);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
int main() {
|
|||
|
|
// 输入n个城镇，m条路径,S:起点，T:终点
|
|||
|
|
cin >> n >> m >> S >> T;
|
|||
|
|
// 初始化邻接表
|
|||
|
|
memset(h, -1, sizeof h);
|
|||
|
|
// 读入m条路径
|
|||
|
|
for (int i = 0; i < m; i++) {
|
|||
|
|
// 每条路径的起点，终点，花费
|
|||
|
|
int a, b, c;
|
|||
|
|
cin >> a >> b >> c;
|
|||
|
|
// 无向边
|
|||
|
|
add(a, b, c), add(b, a, c);
|
|||
|
|
}
|
|||
|
|
spfa();
|
|||
|
|
// 输出终点T的单源最短距离
|
|||
|
|
printf("%d\n", d[T]);
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|