# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1113116 | ortsac | Rainforest Jumps (APIO21_jumps) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#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;
vector<vector<pii>> adj;
vector<bool> vis;
void dfs(int node) {
if (vis[node]) return;
vis[node] = 1;
for (auto u : adj[node]) dfs(u.fr);
}
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) {
adj.clear();
vis.clear();
adj.resize(n);
vis.resize(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));
priority_queue<pair<double, pii>> pq;
dist[0][0] = 0;
pq.push({0, {0, 0}});
for (int i = 0; i < n; i++) {
if ((arr[i] == 0) && vis[i]) {
dist[i][0] = 0;
pq.push({0, {i, 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;
}