#include <bits/stdc++.h>
#include "cyberland.h"
using namespace std;
const int MAX_N = 1e5 + 5;
const int MAX_K = 35;
const long long INF = 1e18 + 7;
vector <pair <int, int>> graph[MAX_N];
double dist[MAX_N][MAX_K];
bool visited[MAX_N][MAX_K];
double solve(int N, int M, int K, int H, vector <int> x, vector <int> y, vector <int> c, vector <int> arr) {
for (int i = 0; i < N; i++) {
graph[i].clear();
for (int k = 0; k <= K; k++) {
dist[i][k] = INF;
visited[i][k] = false;
}
}
for (int i = 0; i < M; i++) {
graph[x[i]].emplace_back(y[i], c[i]);
graph[y[i]].emplace_back(x[i], c[i]);
}
priority_queue <tuple <double, int, int>> pq;
pq.emplace(0, 0, 0);
dist[0][0] = 0;
while (!pq.empty()) {
auto [d, k, u] = pq.top();
pq.pop();
if (visited[u][k] == true) continue;
visited[u][k] = true;
for (auto [v, c] : graph[u]) {
if (v == H) continue;
if (visited[v][k] == false and dist[u][k] + c < dist[v][k]) {
dist[v][k] = dist[u][k] + c;
pq.emplace(-dist[v][k], k, v);
}
if (arr[v] == 0 and visited[v][k] == false and 0 < dist[v][k]) {
dist[v][k] = 0;
pq.emplace(0, k, v);
}
else if (arr[v] == 2 and k < K and visited[v][k + 1] == false and (dist[u][k] + c) / 2 < dist[v][k + 1]) {
dist[v][k + 1] = (dist[u][k] + c) / 2;
pq.emplace(-dist[v][k + 1], k + 1, v);
}
}
}
double res = INF;
for (int k = 0; k <= K; k++) {
for (auto [v, c] : graph[H]) {
res = min(res, dist[v][k] + c);
}
}
if (res == INF) res = -1;
return res;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
20 ms |
2816 KB |
Correct. |
2 |
Correct |
20 ms |
2836 KB |
Correct. |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
21 ms |
3112 KB |
Correct. |
2 |
Correct |
24 ms |
3108 KB |
Correct. |
3 |
Correct |
23 ms |
3144 KB |
Correct. |
4 |
Correct |
30 ms |
3084 KB |
Correct. |
5 |
Correct |
25 ms |
3216 KB |
Correct. |
6 |
Correct |
26 ms |
6428 KB |
Correct. |
7 |
Correct |
29 ms |
6344 KB |
Correct. |
8 |
Correct |
18 ms |
10124 KB |
Correct. |
9 |
Correct |
24 ms |
2772 KB |
Correct. |
10 |
Correct |
29 ms |
2772 KB |
Correct. |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
30 ms |
3088 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
73 ms |
24196 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
20 ms |
3156 KB |
Correct. |
2 |
Correct |
27 ms |
3248 KB |
Correct. |
3 |
Correct |
23 ms |
3140 KB |
Correct. |
4 |
Correct |
28 ms |
6400 KB |
Correct. |
5 |
Correct |
20 ms |
2772 KB |
Correct. |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
25 ms |
3140 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
244 ms |
3640 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
3069 ms |
12640 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |