#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pii pair<long long, long long>
#define fr first
#define se second
const double INF = 1e100;
double solve(int32_t n, int32_t m, int32_t k, int32_t h, vector<int32_t> x, vector<int32_t> y, vector<int32_t> c, vector<int32_t> arr) {
vector<vector<pii>> adj(n);
for (int i = 0; i < m; i++) {
int a = x[i], b = y[i], co = c[i];
if (a != h) adj[a].push_back({b, co});
if (b != h) adj[b].push_back({a, co});
}
vector<vector<bool>> prop(n, vector<bool>(31));
vector<vector<double>> dist(n, vector<double>(31, INF));
dist[0][0] = 0;
priority_queue<pair<double, pii>> pq;
pq.push({0, {0, 0}});
while (!pq.empty()) {
auto [u, qtd] = pq.top().se;
pq.pop();
if (prop[u][qtd]) continue;
prop[u][qtd] = 1;
for (auto [v, cost] : adj[u]) {
// propaga sem aumentar o k
if (!prop[v][qtd]) {
double novo = (((double)cost) + dist[u][qtd]);
if (arr[v] == 0) novo = 0;
if (novo < dist[v][qtd]) {
dist[v][qtd] = novo;
pq.push({-novo, {v, qtd}});
}
}
if ((qtd < k) && (arr[v] == 2) && (!prop[v][qtd + 1])) {
double novo = (((double)cost) + dist[u][qtd]) / 2.0;
if (novo < dist[v][qtd + 1]) {
dist[v][qtd + 1] = novo;
pq.push({-novo, {v, qtd + 1}});
}
}
}
}
double ans = INF;
for (int i = 0; i <= k; i++) {
ans = min(ans, dist[h][i]);
}
if (ans == INF) return -1;
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
19 ms |
592 KB |
Correct. |
2 |
Correct |
19 ms |
596 KB |
Correct. |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
25 ms |
848 KB |
Correct. |
2 |
Correct |
31 ms |
932 KB |
Correct. |
3 |
Correct |
34 ms |
896 KB |
Correct. |
4 |
Correct |
31 ms |
916 KB |
Correct. |
5 |
Correct |
33 ms |
968 KB |
Correct. |
6 |
Correct |
28 ms |
4884 KB |
Correct. |
7 |
Correct |
35 ms |
4848 KB |
Correct. |
8 |
Correct |
18 ms |
9296 KB |
Correct. |
9 |
Correct |
31 ms |
336 KB |
Correct. |
10 |
Correct |
26 ms |
336 KB |
Correct. |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
31 ms |
872 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
64 ms |
26440 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
26 ms |
912 KB |
Correct. |
2 |
Correct |
34 ms |
864 KB |
Correct. |
3 |
Correct |
29 ms |
880 KB |
Correct. |
4 |
Correct |
29 ms |
4524 KB |
Correct. |
5 |
Correct |
24 ms |
336 KB |
Correct. |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
27 ms |
828 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
199 ms |
1368 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
2 ms |
848 KB |
Execution killed with signal 11 |
2 |
Halted |
0 ms |
0 KB |
- |