oj.uz

Submission #983165

# Submission time Handle Problem Language Result Execution time Memory
983165 2024-05-15T08:51:47 Z feev1x Cyberland (APIO23_cyberland) C++17
13 / 100
 3000 ms 7508 KB 
#include "cyberland.h"

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

const double INF = numeric_limits<double>::max();

double nonol(vector<vector<pair<int, int>>> g, int N, int M, int K, int H, vector<int> arr) {
	vector<bool> used(N), ot(N);
	function<void(int, int)> dfs = [&](int v, int cnt) {
		assert(!used[v]);
		used[v] = true;
		for (auto [to, weight]: g[v]) {
			if (!used[to]) {
				if (arr[to] == 2 && cnt) {
					ot[to] = true;
					dfs(to, cnt - 1);
				} else dfs(to, cnt);
			}
		}
	};
	vector<double> d(N + 1, INF);
	d[0] = 0;
	set<pair<double, int>> q;
	q.emplace(d[0], 0);
	while (!q.empty()) {
		int nearest = q.begin()->second;
		q.erase(q.begin());
		for (auto [to, weight]: g[nearest]) {
			double dis = d[nearest] + weight;
			if (ot[to]) dis /= 2;
			if (!arr[to]) dis = 0;
			if (d[to] > dis) {
				q.erase(pair{d[to], to});
				d[to] = dis;
				q.emplace(d[to], to);
			}
		}
	}
        if (d[H] == INF) d[H] = -1;
	return d[H];
}

double s1(vector<vector<pair<int, int>>> g, int N, int M, int K, int H, vector<int> arr) {
	double dis = INF;
	bool can = false;
	double zap;
	int u;
	for (auto [to, weight]: g[0]) {
		if (to == H) {
			dis = fmin(dis, weight);
		} else can = true, zap = weight, u = to;
	}
	if (can) {
		double w;
		can = false;
		for (auto [to, weight]: g[u]) {
			if (to == H) can = true, w = weight;
		}
		if (can) {
			if (arr[u] == 0) dis = fmin(dis, w);
			else if (arr[u] == 1) dis = fmin(dis, w + zap);
			else dis = fmin(dis, zap / 2. + w);
		}
	}
	if (dis == INF) dis = -1;
	return dis;
}

double s4(std::vector<int> x, std::vector<int> y, std::vector<int> c, int N, int M, int K, int H, vector<int> arr) {
	vector<bool> ot(N);
	while(true)N++;
	int cnt = K;
	for (int i = H - 1; i > 0; --i) {
		if (arr[i] == 2 && cnt) cnt--, ot[i] = true;
	}
	cnt = K;
	for (int i = H + 1; i < N; ++i) {
		if (arr[i] == 2 && cnt) cnt--, ot[i] = true;
	}
	double dis = INF;
	double tmp = 0;
	for (int i = 0; i < H; ++i) {
		tmp += c[i];
		if (arr[i + 1] == 0) tmp = 0;
		if (arr[i + 1] == 2) tmp /= 2;
	}
	dis = tmp;
	tmp = 0;
	bool flag = false;
	for (int i = N - 1; i > H; --i) {
		tmp += c[i - 1];
		if (arr[i - 1] == 0) tmp = 0, flag = true;
		if (arr[i - 1] == 2) tmp /= 2;
	}
	if (flag) dis = min(dis, tmp);
	if (dis == INF) dis = -1;
	return dis;
}
double yesnol(vector<vector<pair<int, int>>> g, int N, int M, int K, int H, vector<int> arr) {return -1;}

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) {
    vector<vector<pair<int, int>>> g(N);
    bool is2 = true;
    for (int i = 0; i < M; ++i) {
		g[x[i]].emplace_back(y[i], c[i]);
		g[y[i]].emplace_back(x[i], c[i]);
		if (x[i] != i || y[i] != i + 1) is2 = false;
	}
	if (N <= 3) return s1(g, N, M, K, H, arr);
	else if (is2) return s4(x, y, c, N, M, K, H, arr);
	else if (find(arr.begin(), arr.end(), 0LL) == arr.end()) return nonol(g, N, M, K, H, arr);
	else return yesnol(g, N, M, K, H, arr);
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 860 KB Correct.
2 Correct 14 ms 856 KB Correct.

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 1356 KB Correct.
2 Correct 31 ms 1620 KB Correct.
3 Correct 35 ms 1620 KB Correct.
4 Correct 27 ms 1628 KB Correct.
5 Correct 27 ms 1620 KB Correct.
6 Correct 25 ms 2788 KB Correct.
7 Correct 30 ms 3016 KB Correct.
8 Correct 13 ms 4200 KB Correct.
9 Correct 27 ms 1368 KB Correct.
10 Correct 27 ms 1432 KB Correct.

Subtask #3
0.0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 19 ms 1628 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #4
0.0 / 19.0

# Verdict Execution time Memory Grader output
1 Execution timed out 3055 ms 7508 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #5
0.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 1372 KB Correct.
2 Correct 25 ms 1620 KB Correct.
3 Correct 24 ms 1624 KB Correct.
4 Correct 23 ms 2652 KB Correct.
5 Execution timed out 3032 ms 344 KB Time limit exceeded
6 Halted 0 ms 0 KB -

Subtask #6
0.0 / 16.0

# Verdict Execution time Memory Grader output
1 Incorrect 19 ms 1628 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #7
0.0 / 29.0

# Verdict Execution time Memory Grader output
1 Execution timed out 3048 ms 1112 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #8
0.0 / 3.0

# Verdict Execution time Memory Grader output
1 Execution timed out 3022 ms 1116 KB Time limit exceeded
2 Halted 0 ms 0 KB -