제출 #756680

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
756680		vioalbert	사이버랜드 (APIO23_cyberland)	C++17	15 / 100	383 ms	9060 KiB

cyberland

#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;

const double INF = 1e15;

double solve(int N, int M, int K, int H, vector<int> x,
             vector<int> y, vector<int> c, vector<int> arr) {
  vector adj(N, vector<pair<int, int>>());
  for(int i = 0; i < M; i++) {
    adj[x[i]].emplace_back(y[i], c[i]);
    adj[y[i]].emplace_back(x[i], c[i]);
  }
  vector<int> mark(N, 0);
  const auto connected = [&](const auto &self, int u, int target) -> bool {
    if(u == target) return true;
    mark[u] = 1;
    for(auto [v, c] : adj[u]) if(!mark[v]) {
      if(self(self, v, target)) return true;
    }
    return false;
  };
  if(!connected(connected, 0, H)) {
    return -1.0;
  }
  const auto dijkstra = [&](const vector<int> &sources, vector<double> &dist) {
    using state = pair<double, int>; // dist, node
    vector<int> vis(N, 0);
    priority_queue<state, vector<state>, greater<state>> pq;
    for(int i : sources) pq.push({dist[i], i});
    while(!pq.empty()) {
      int u = pq.top().second; pq.pop();
      if(vis[u]) continue;
      vis[u] = 1;
      for(auto [v, c] : adj[u]) {
        if(dist[v] > dist[u] + c) {
          pq.push({dist[v] = dist[u] + c, v});
        }
      }
    }
  };
  vector<int> sources{0};
  for(int i = 1; i < N; i++) if(arr[i] == 0)
    sources.push_back(i);
  vector<double> dist(N, INF);
  for(int i : sources)
    dist[i] = 0.0;
  vector<double> new_dist(N);
  double ans = INF;
  for(int it = 0; it < min(K, 95); it++) {
    // cerr << "it = " << it << '\n';
    if(it == 1) {
      sources.clear();
      for(int i = 0; i < N; i++) if(arr[i] != 1)
        sources.push_back(i);
    }
    if(it > 0) {
      for(int i = 0; i < N; i++) {
        if(arr[i] == 0) new_dist[i] = 0.0;
        else if(arr[i] == 1) new_dist[i] = INF;
        else {
          new_dist[i] = INF;
          for(auto [v, c] : adj[i])
            new_dist[i] = min(new_dist[i], (dist[v] + c) / 2.0);
        }
      }
      swap(dist, new_dist);
    }
    dijkstra(sources, dist);
    ans = min(ans, dist[H]);
    // for(int i = 0; i < N; i++) {
    //   cerr << "dist[" << i << "] = " << dist[i] << '\n';
    // }
  }
  return ans;
}

• Subtask 1: 0 / 5
• Subtask 2: 8 / 8
• Subtask 3: 0 / 13
• Subtask 4: 0 / 19
• Subtask 5: 7 / 7
• Subtask 6: 0 / 16
• Subtask 7: 0 / 29
• Subtask 8: 0 / 3