제출 #919138

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

cyberland

#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<double, 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));
	std::vector vis(N, std::vector<bool> (K + 1, false));
	while(!pq.empty()) {
		auto [cost, node, bound] = pq.top();
		pq.pop();
		if(vis[node][bound]) {
			continue;
		}

		dist[node][bound] = cost;
		vis[node][bound] = true;
		
		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);
			}
		}
	}

	return -1;
}

• 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