#include <bits/stdc++.h>
using namespace std;
typedef long long i64;
const int INF = 1000000007;
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]) {
vector<vector<pair<int, i64>>> edges(N);
for (int i = 0; i < M; ++i) {
edges[R[i][1]].emplace_back(R[i][0], i64(L[i]));
edges[R[i][0]].emplace_back(R[i][1], i64(L[i]));
}
vector<vector<int>> sp_edges(N);
vector<bool> visited(N, false);
vector<i64> sp1(N, 1e18);
priority_queue<array<i64, 3>> q;
for (int i = 0; i < K; ++i) {
sp1[P[i]] = 0;
q.push({0, P[i], P[i]});
}
while (!q.empty()) {
i64 dist = -q.top()[0], node = q.top()[1], prev = q.top()[2];
q.pop();
if (node != prev && dist == sp1[node]) sp_edges[node].push_back(prev);
if (visited[node]) continue;
visited[node] = true;
for (auto [next, weight] : edges[node]) {
if (dist + weight <= sp1[next]) {
sp1[next] = dist + weight;
q.push({-sp1[next], next, node});
}
}
}
fill(visited.begin(), visited.end(), false);
vector<i64> sp2(N, 1e18);
q.push({0, 0, 0});
while (!q.empty()) {
i64 dist = -q.top()[0], node = q.top()[1], prev = q.top()[2];
q.pop();
if (visited[node]) continue;
visited[node] = true;
for (auto [next, weight] : edges[node]) {
if (sp_edges[node].size() == 1 && next == sp_edges[node][0]) continue;
if (dist + weight <= sp2[next]) {
sp2[next] = dist + weight;
q.push({-sp2[next], next, node});
}
}
}
i64 ans = 1e18;
for (int i = 0; i < K; ++i) {
ans = min(ans, sp2[P[i]]);
}
return ans;
}
Compilation message
crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:37:48: warning: unused variable 'prev' [-Wunused-variable]
37 | i64 dist = -q.top()[0], node = q.top()[1], prev = q.top()[2];
| ^~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |