Submission #983958

#	Time	Username	Problem	Language	Result	Execution time	Memory
983958		Halym2007	Cyberland (APIO23_cyberland)	C++17	21 / 100	26 ms	8024 KiB

cyberland

#include <bits/stdc++.h>
#include "cyberland.h"
using namespace std;
#define ff first
#define ss second
#define pb push_back
#define sz size()
#define ll long long
const int N = 1e5 + 5;
 
vector <pair <int, ll>> v[N];
 
ll dis[N];
bool vis[N];
 
void dfs (int x) {
	vis[x] = 1;
	for (pair <int, ll> i : v[x]) {
		if (vis[i.ff]) continue;
		dfs (i.ff);
	}
}
 
double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
 	bool subtask = 0;
	for (int i = 0; i < M; ++i) {
		v[x[i]].pb ({y[i], (ll)c[i]});
		v[y[i]].pb ({x[i], (ll)c[i]});
		if (arr[i] == 2) {
			subtask = 1;
		}
	}
	double jogap;
	if (!subtask) {
		for (int i = 0; i < N; ++i) {
			dis[i] = 1e18;
		}
		vis[H] = 1;
		dfs (0);
		vis[H] = 0;
		priority_queue <pair <ll, int>, vector <pair <ll, int>>, greater <pair <ll, int>>> q;
		q.push ({0, H});
		dis[H] = 0;
		while (!q.empty()) {
			int x = q.top().ss;
			q.pop();
			for (pair <int, ll> i : v[x]) {
				if (dis[i.ff] > dis[x] + i.ss) {
					dis[i.ff] = dis[x] + i.ss;
					q.push({dis[i.ff], i.ff});
				}
			}
		}
		if (dis[0] != 1e18) {
			jogap = dis[0];
			for (int i = 1; i < N; ++i) {
				if (!arr[i] and vis[i]) jogap = min (jogap, (double)dis[i]);
			}
		}
		else jogap = -1;
	}
	else {
		int sana = 0;
		for (int i = H - 1; i >= 0; i--) {
			if (!arr[i]) break;
			if (arr[i] == 2 and sana < K) {
				sana++;
				vis[i] = 1;
			}
		}
		for (int i = 0; i < H; ++i) {
			double jj = c[i];
			jogap += jj;
			if (!arr[i]) jogap = 0;
			else if (vis[i]) {
				jogap /= 2.0;
			}
		}
	}
	for (int i = 0; i < N; ++i) {
		v[i].clear();
		vis[i] = 0;
	}
	return jogap;
}
 
//int main() {
//	freopen ("input.txt", "r", stdin);
//	int T;
//	assert(1 == scanf("%d", &T));
//	while (T--) {
//    	int N,M,K,H;
//    	assert(4 == scanf("%d %d %d\n%d", &N, &M, &K, &H));
//	    std::vector<int> x(M);
//	    std::vector<int> y(M);
//	    std::vector<int> c(M);
//	    std::vector<int> arr(N);
//    	for (int i=0;i<N;i++)
//      		assert(1 == scanf("%d", &arr[i]));
//		for (int i=0;i<M;i++)
//    		assert(3 == scanf("%d %d %d", &x[i], &y[i], &c[i]));
//		printf("%.12lf\n", solve(N, M, K, H, x, y, c, arr));
//	}
//}

Compilation message (stderr)

cyberland.cpp: In function 'double solve(int, int, int, int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
cyberland.cpp:73:10: warning: 'jogap' may be used uninitialized in this function [-Wmaybe-uninitialized]
   73 |    jogap += jj;
      |    ~~~~~~^~~~~

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