Submission #649101

#	Time	Username	Problem	Language	Result	Execution time	Memory
649101		ymm	The Big Prize (IOI17_prize)	C++17	100 / 100	49 ms	5228 KiB

prize

#include "prize.h"
#include <bits/stdc++.h>
#define Loop(x,l,r) for (ll x = (l); x < (ll)(r); ++x)
#define LoopR(x,l,r) for (ll x = (r)-1; x >= (ll)(l); --x)
typedef long long ll;
typedef std::pair<int, int> pii;
typedef std::pair<ll , ll > pll;
using namespace std;

const int N = 200'010;

jmp_buf ret_env, retry_env;
int ret_val;
int cnt_expen = N;
auto ret_ask(int i)
{
	static vector<int> memoize[N];
	vector<int> v;
	if (memoize[i].size())
		v = memoize[i];
	else
		v = memoize[i] = ask(i);
	if (v[0] + v[1] == 0) {
		ret_val = i;
		longjmp(ret_env, 1);
	}
	if (v[0] + v[1] > cnt_expen) {
		cnt_expen = v[0] + v[1];
		longjmp(retry_env, 1);
	}
	return v;
}


void solve(int l, int r, int cntl, int cnt, int cntr)
{
	if (cnt == 0 || l >= r)
		return;
	int ml = (l+r)/2, mr = (l+r)/2;
	vector<int> lst;
	int lstd;
	while (l < ml || mr < r) {
		if (mr < r) {
			lstd = 1;
			lst = ret_ask(mr++);
			if (lst[0] + lst[1] == cnt_expen)
				break;
		}
		if (l < ml) {
			lstd = 0;
			lst = ret_ask(--ml);
			if (lst[0] + lst[1] == cnt_expen)
				break;
		}
	}
	if (l == ml && r == mr)
		return;
	int to_l = lstd == 1? mr-ml-1: 0;
	int to_r = lstd == 0? mr-ml-1: 0;
	solve(l, ml, cntl, lst[0] - cntl - to_l, lst[1] + to_l);
	solve(mr, r, lst[0] + to_r, lst[1] - cntr - to_r, cntr);
}

int find_best(int n)
{
	if (setjmp(ret_env))
		return ret_val;
	if (n <= 5000) {
		Loop (i,0,n)
			ret_ask(i);
	}
	cnt_expen = 1;
	setjmp(retry_env);
	solve(0, n, 0, cnt_expen, 0);
	return -1;
}

Compilation message (stderr)

prize.cpp: In function 'void solve(int, int, int, int, int)':
prize.cpp:58:22: warning: 'lstd' may be used uninitialized in this function [-Wmaybe-uninitialized]
   58 |  int to_l = lstd == 1? mr-ml-1: 0;
      |             ~~~~~~~~~^~~~~~~~~~~~

• Subtask 1: 20 / 20
• Subtask 2: 80 / 80