题目地址:
https://www.acwing.com/problem/content/1134/
N N N头牛要去参加在某农场举行的一场编号为 X X X的牛的派对。有 M M M条有向道路,每条路长 T i T_i Ti;每头牛参加完派对后都必须回到家,每头牛都会选择最短路径。求这 N N N头牛的最短路径(一个来回)中最长的一条的长度。特别提醒:可能有权值不同的重边。
输入格式:
第一行包含三个整数
N
,
M
,
X
N,M,X
N,M,X。接下来
M
M
M行,每行包含三个整数
A
i
,
B
i
,
T
i
A_i,B_i,T_i
Ai,Bi,Ti,表示有一条从
A
i
A_i
Ai到
B
i
B_i
Bi的路,长度为
T
i
T_i
Ti。
输出格式:
共一行,一个数,表示最短路中最长的一条的长度。
数据范围:
1
≤
N
≤
1000
1≤N≤1000
1≤N≤1000
1
≤
X
≤
N
1≤X≤N
1≤X≤N
1
≤
M
≤
1
0
5
1≤M≤10^5
1≤M≤105
1
≤
T
i
≤
100
1≤T_i≤100
1≤Ti≤100
可以建正向图和反向图来两遍Dijkstra算法。代码如下:
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
using PII = pair<int, int>;
const int N = 1010, M = 2e5 + 10;
int n, m, X;
int h[2][N], e[M], ne[M], w[M], idx;
int dist[2][N];
bool vis[N];
void add(int h[], int a, int b, int c) {
e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}
void dijkstra(int hi, int di) {
memset(vis, 0, sizeof vis);
dist[di][X] = 0;
priority_queue<PII, vector<PII>, greater<>> heap;
heap.push({0, X});
while (heap.size()) {
auto t = heap.top(); heap.pop();
int v = t.second;
if (vis[v]) continue;
vis[v] = true;
for (int i = h[hi][v]; ~i; i = ne[i]) {
int j = e[i];
if (!vis[j] && dist[di][j] > dist[di][v] + w[i]) {
dist[di][j] = dist[di][v] + w[i];
heap.push({dist[di][j], j});
}
}
}
}
int main() {
scanf("%d%d%d", &n, &m, &X);
memset(h, -1, sizeof h);
while (m--) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(h[0], a, b, c), add(h[1], b, a, c);
}
memset(dist, 0x3f, sizeof dist);
dijkstra(0, 0);
dijkstra(1, 1);
int res = 0;
for (int i = 1; i <= n; i++)
res = max(res, dist[0][i] + dist[1][i]);
printf("%d\n", res);
return 0;
}
时间复杂度 O ( M log N ) O(M\log N) O(MlogN),空间 O ( N ) O(N) O(N)。