Submission #919131

#TimeUsernameProblemLanguageResultExecution timeMemory
919131MisterReaperCyberland (APIO23_cyberland)C++17
0 / 100
31 ms22352 KiB
#include <bits/stdc++.h>
//#include "cyberland.h"
using i64 = long long;

struct DSU {
  int n;
  std::vector<int> par;
  DSU(int _n) : n(_n) {
    par.resize(n);
    std::iota(par.begin(), par.end(), 0);
  }

  int get(int x) {
    if(par[x] == x) {
      return x;
    }

    return par[x] = get(par[x]);
  }

  bool same(int a, int b) {
    return get(a) == get(b);
  }

  bool unite(int u, int v) {
    if(same(u, v)) {
      return false;
    }

    u = get(u);
    v = get(v);
    par[u] = v;

    return true;
  }
};

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) {
  DSU dsu(N);
  std::vector<std::pair<int, int>> adj[N];
  for(int i = 0; i < M; i++) {
    dsu.unite(x[i], y[i]);
    adj[x[i]].emplace_back(y[i], c[i]);
    adj[y[i]].emplace_back(x[i], c[i]);
  }

  if(!dsu.same(0, H)) {
    return -1;
  }

  K = std::min(100, K);

  std::vector<double> pw(K + 1, 1);
  for(int i = 1; i <= K; i++) {
    pw[i] = pw[i - 1] / 2;
  }

  using T = std::tuple<int, int, int>;
  std::priority_queue<T, std::vector<T>, std::greater<T>> pq;
  pq.emplace(0, H, 0);

  std::vector dist(N, std::vector<double> (K + 1, -1));
  while(!pq.empty()) {
    auto [cost, node, bound] = pq.top();
    pq.pop();
    if(dist[node][bound] != -1) {
      continue;
    }

    dist[node][bound] = cost;

    if(arr[node] == 0) {
      return cost;
    } else if(node == 0) {
      return cost;
    }

    for(auto [child, w] : adj[node]) {
      pq.emplace(cost + w * pw[bound], child, bound);
      if(arr[child] == 2 && bound + 1 <= K) {
        pq.emplace(cost + w * pw[bound + 1], child, bound + 1);
      }
    }
  }

  assert(false);
  return -1;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...