#include "cyberland.h"
#include <bits/stdc++.h>
const int MAX_N = 1e5 + 10;
const int MAX_K = 100;
typedef long double ld;
ld dist[MAX_N][MAX_K];
std::vector<std::pair<int, int> > g[MAX_N];
bool used[MAX_N];
void dfs(int x) {
if(used[x]) { return; }
used[x] = true;
for(const auto &it : g[x]) {
dfs(it.first);
}
}
double solve(int N, int M, int K, int H, std::vector<int> x, std::vector<int> y, std::vector<int> c, std::vector<int> arr) {
for(int i = 0; i < M; i ++) {
g[x[i]].push_back({y[i], c[i]});
g[y[i]].push_back({x[i], c[i]});
}
dfs(0);
std::priority_queue<std::pair<ld, std::pair<int, int> > > pq;
K = std::min(K, MAX_K - 1);
for(int i = 0; i < N; i ++) {
for(int j = 0; j <= K; j ++) {
dist[i][j] = 1e18;
}
}
for(int i = 0; i < N; i ++) {
if(arr[i] == 0 || i == 0) {
dist[i][arr[i]] = 0;
pq.push({0, {arr[i] != 0, i}});
}
}
while(!pq.empty()) {
auto curr = pq.top(); pq.pop();
curr.first *= -1;
if(curr.first > dist[curr.second.second][curr.second.first]) {
continue;
}
for(const auto &it : g[curr.second.second]) {
ld new_dist = curr.first + it.second;
if(new_dist < dist[it.first][curr.second.first]) {
dist[it.first][curr.second.first] = new_dist;
pq.push({-new_dist, {curr.second.first, it.first}});
}
if(arr[it.first] == 2 && curr.second.first < K) {
new_dist /= 2;
if(new_dist < dist[it.first][curr.second.first + 1]) {
dist[it.first][curr.second.first + 1] = new_dist;
pq.push({-new_dist, {curr.second.first + 1, it.first}});
}
}
}
}
ld ret = 1e18;
for(int i = 0; i <= K; i ++) {
ret = std::min(ret, dist[N - 1][i]);
}
return ret;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
3069 ms |
3024 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1295 ms |
19728 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1085 ms |
12300 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1597 ms |
131900 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
661 ms |
12064 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
823 ms |
12648 KB |
Wrong Answer. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
3067 ms |
19492 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
3057 ms |
53352 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |