#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
#define INF (long long)(1e18 + 1)
#define int long long
int n, m;
const int maxn = 1e5 + 69;
vector <pair<int, int>> adj[maxn];
int dp[maxn][2];
int32_t travel_plan(int32_t N, int32_t M, int32_t R[][2], int32_t L[], int32_t K, int32_t P[])
{
n = N, m = M;
for (int i = 0; i < n; i++){
dp[i][0] = dp[i][1] = INF;
}
for (int i = 0; i < m; i++){
adj[R[i][0]].push_back({R[i][1], L[i]});
adj[R[i][1]].push_back({R[i][0], L[i]});
}
for (int i = 0; i < K; i++){
dp[P[i]][0] = dp[P[i]][1] = 0;
}
//dp[i][0] = best path, dp[i][1] = second best path, update only with help of second best path values
priority_queue <pair<int, int>> pq;
for (int i = 0; i < n; i++){
if (dp[i][1] != INF)
pq.push({-dp[i][1], i});
}
while (!pq.empty()){
auto pi = pq.top();
pq.pop();
int d = -pi.first;
int u = pi.second;
if (d != dp[u][1]) continue;
for (pair <int, int> pi : adj[u]){
int v = pi.first;
int w = pi.second;
int val = d + w;
if (val < dp[v][0]){
int temp = dp[v][1];
dp[v][1] = dp[v][0];
dp[v][0] = val;
if (dp[v][1] != temp)
pq.push({-dp[v][1], v});
} else if (val < dp[v][1]) {
dp[v][1] = val;
pq.push({-dp[v][1], v});
}
}
}
return dp[0][1];
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
2644 KB |
Output is correct |
2 |
Correct |
2 ms |
2644 KB |
Output is correct |
3 |
Correct |
1 ms |
2644 KB |
Output is correct |
4 |
Correct |
2 ms |
2772 KB |
Output is correct |
5 |
Correct |
2 ms |
2772 KB |
Output is correct |
6 |
Correct |
2 ms |
2644 KB |
Output is correct |
7 |
Correct |
2 ms |
2772 KB |
Output is correct |
8 |
Correct |
2 ms |
2772 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
2644 KB |
Output is correct |
2 |
Correct |
2 ms |
2644 KB |
Output is correct |
3 |
Correct |
1 ms |
2644 KB |
Output is correct |
4 |
Correct |
2 ms |
2772 KB |
Output is correct |
5 |
Correct |
2 ms |
2772 KB |
Output is correct |
6 |
Correct |
2 ms |
2644 KB |
Output is correct |
7 |
Correct |
2 ms |
2772 KB |
Output is correct |
8 |
Correct |
2 ms |
2772 KB |
Output is correct |
9 |
Correct |
3 ms |
3028 KB |
Output is correct |
10 |
Correct |
1 ms |
2644 KB |
Output is correct |
11 |
Correct |
2 ms |
2752 KB |
Output is correct |
12 |
Correct |
4 ms |
3284 KB |
Output is correct |
13 |
Correct |
5 ms |
3284 KB |
Output is correct |
14 |
Correct |
1 ms |
2644 KB |
Output is correct |
15 |
Correct |
2 ms |
2772 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
2644 KB |
Output is correct |
2 |
Correct |
2 ms |
2644 KB |
Output is correct |
3 |
Correct |
1 ms |
2644 KB |
Output is correct |
4 |
Correct |
2 ms |
2772 KB |
Output is correct |
5 |
Correct |
2 ms |
2772 KB |
Output is correct |
6 |
Correct |
2 ms |
2644 KB |
Output is correct |
7 |
Correct |
2 ms |
2772 KB |
Output is correct |
8 |
Correct |
2 ms |
2772 KB |
Output is correct |
9 |
Correct |
3 ms |
3028 KB |
Output is correct |
10 |
Correct |
1 ms |
2644 KB |
Output is correct |
11 |
Correct |
2 ms |
2752 KB |
Output is correct |
12 |
Correct |
4 ms |
3284 KB |
Output is correct |
13 |
Correct |
5 ms |
3284 KB |
Output is correct |
14 |
Correct |
1 ms |
2644 KB |
Output is correct |
15 |
Correct |
2 ms |
2772 KB |
Output is correct |
16 |
Correct |
329 ms |
66728 KB |
Output is correct |
17 |
Correct |
54 ms |
14276 KB |
Output is correct |
18 |
Correct |
96 ms |
16668 KB |
Output is correct |
19 |
Correct |
426 ms |
72864 KB |
Output is correct |
20 |
Correct |
237 ms |
55020 KB |
Output is correct |
21 |
Correct |
28 ms |
8228 KB |
Output is correct |
22 |
Correct |
241 ms |
49920 KB |
Output is correct |