제출 #1340073

#제출 시각아이디문제언어결과실행 시간메모리
1340073limits사이버랜드 (APIO23_cyberland)C++20
44 / 100
3096 ms2162688 KiB
#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3,unroll-loops")

#include <bits/stdc++.h>

using namespace std;

#define f0r(i, n) for (auto i = 0; i < (n); ++i)
#define fnr(i, n, k) for (auto i = (n); i < (k); ++i)
#define all(v) (v).begin(), (v).end()
#define pb push_back
#define F first
#define S second
#define ctn(x) cout << x << '\n'
#define forl(a, l) for (auto a : l)
#define ctl(l) for (auto &a : (l)) cout << a << ' '; cout << endl;
#define lb(v, x) (lower_bound(all(v), x) - begin(v))
#define ub(v, x) (upper_bound(all(v), x) - begin(v))
#define pq priority_queue

template <class T>
using V = vector<T>;
using ll = long long;
using vi = V<int>;
using ld = long double;
using vl = V<ld>;
using pi = pair<int, int>;
using pl = pair<ld, int>;
using ti = tuple<int, int, int>;
using Adj = V<V<pi>>;
using vvi = V<vi>;

#include "cyberland.h"

Adj G;

V<vl> djikstra(int s, int K, vi &A) {
	V<vl> dist(G.size(), vl(K+1, 1e15));

	using pr = pair<ld, pi>;
	priority_queue<pr, V<pr>, greater<pr>> pq;

	pq.push({0, {s, 0}});
	dist[s][0] = 0;

	while (!pq.empty()) {
		auto [d, p] = pq.top();
		auto [u, k] = p;
		pq.pop();
		if (d > dist[u][k]) continue;

		for (auto [c, w]: G[u]) {
			if (d + w < dist[c][k]) {
				dist[c][k] = d+w;
				pq.push({d+w, {c, k}});
			}
			if (A[c] == 2 && k < K && (d+w)/2 < dist[c][k+1]) {
				dist[c][k+1] = (d+w)/2;
				pq.push({dist[c][k+1], {c, k+1}});
			}
		}
	}
	return dist;
}

double solve(int N, int M, int K, int H, vi x, vi y, vi c, vi arr) {
	G.assign(N, {});
	f0r(i, M) {
		G[x[i]].pb({y[i], c[i]});
		G[y[i]].pb({x[i], c[i]});
	}
	V<vl> dist = djikstra(H, K, arr);

	queue<int> Q;
	Q.push(0);
	V<bool> vis(N);
	vis[H] = 1;
	while (!Q.empty()) {
		int u = Q.front();
		Q.pop();
		if (vis[u]) continue;
		vis[u] = 1;
		for (auto [c, w]: G[u]) {
			Q.push(c);
		}
	}

	arr[0] = 0;
	ld mn = 1e15;
	f0r(i, N) if (!arr[i] && vis[i]) mn = min(mn, *min_element(all(dist[i])));
	return mn == 1e15 ? -1 : mn;
}
#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...