题目描述
给一个 n(1≤n≤2500) 个点 m(1≤m≤6200) 条边的无向图,求 s 到 t 的最短路。
输入格式
第一行四个由空格隔开的整数 n、m、s、t。
之后的 m 行,每行三个正整数 s_i、t_i、w_i(1≤wi≤10^9),表示一条从 s_i 到 t_i长度为 w_i 的边。
输出格式
一个整数表示从 s 到 t 的最短路长度。数据保证至少存在一条道路。
样例
输入数据 1
7 11 5 4
2 4 2
1 4 3
7 2 2
3 4 3
5 7 5
7 3 3
6 1 1
6 3 4
2 4 3
5 6 3
7 2 1
输出数据 1
7
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll road[2501][2501];
int main()
{
int n, m, s, t;
bool visit[2501];
ll best[2501] = { 0 };
for (int i = 0; i <= 2500; i++)//init
for (int j = 0; j <= 2500; j++)
road[i][j] = 1e18;
fill(best, best + 2501, 1e18);
cin >> n >> m >> s >> t;
best[s] = 0;
while (m--)
{
int a, b;
ll c;
cin >> a >> b >> c;
if (c<road[a][b])
road[a][b] = road[b][a] = c;
}//input
while (1)//Dijkstra
{
int ma = 1e18, target = -1;
for (int i = 1; i <= n; i++)
{
if (!visit[i] && ma>best[i])
{
ma = best[i];
target = i;
}
}
if (target == -1)break;
visit[target] = 1;
for (int i = 1; i <= n; i++)
{
if (!visit[i])
{
best[i] = min(best[i], best[target] + road[target][i]);
}
}
}
cout << best[t];//output
return 0;
}