Submission #431301

#	Time	Username	Problem	Language	Result	Execution time	Memory
431301		Mazaalai	The Big Prize (IOI17_prize)	C++14	0 / 100	150 ms	864 KiB

prize

#include "prize.h"
#include <bits/stdc++.h>
#define ff first
#define ss second
using namespace std;
typedef pair <int, int> PII;

map <int, PII> asked;
int ans = -1;
bool found;
PII askPos(int n) {
	if (asked.count(n)) return asked[n];
	vector <int> res = ask(n);
	PII cur = {res[0], res[1]};
	if (cur.ff + cur.ss == 0) {
		ans = n;
		found = 1;
	}
	return asked[n] = cur;
}
int middle(PII&a, PII&b) {
	return abs(a.ff - b.ff);
}
int sum(PII a) {
	return a.ff + a.ss;
}
int find_best(int n) {
	found = 0;
	int m = 5, st = 0, end = n - 1, miniCnt = 0;
	PII A, B;
	for (int i = 0; (i < m || n - miniCnt < n / 2) && !found; i++) {
		PII cur = askPos(i);
		int curSum = sum(cur);
		if (curSum > miniCnt) miniCnt = curSum;
		if (curSum == miniCnt) 
			st = i;
	}
	for (int i = n - 1; (i >= 0 || n - miniCnt < n / 2) && !found; i--) {
		PII cur = askPos(i);
		int curSum = sum(cur);
		if (curSum > miniCnt) miniCnt = curSum;
		if (curSum == miniCnt) 
			end = i;
	}
	// cout << miniCnt << '\n';
	int jumped = 0, addPos = 0;
	map <int, int> jumpIdx;

	while(st < end && !found) {
		// cout << st << ' ' << n << '\n';
		A = askPos(st);
		B = askPos(end);
		int l = st, r = end, res = l;
		while(l <= r) {
			int mid = (l+r)/2;
			PII cur = askPos(mid);
			if (A == cur) {
				res = max(res, mid);
				l = mid+1;
			} else {
				r = mid-1;
			}
		}
		st = res+1;
		l = st, r = end, res = end;
		while(l <= r) {
			int mid = (l+r)/2;
			PII cur = askPos(mid);
			if (B == cur) {
				res = min(res, mid);
				r = mid-1;
			} else {
				l = mid+1;
			}
		}
		end = res-1;		
		bool x = 1;
		while(x && !found) {
			x = 0;
			PII cur = askPos(st);
			int curSum = sum(cur);
			if (curSum != miniCnt) {
				x = 1;
				st++;
			}
		}	
		x = 1;
		while(x && !found) {
			x = 0;
			PII cur = askPos(end);
			int curSum = sum(cur);
			if (curSum != miniCnt) {
				x = 1;
				end--;
			}
		}
	}
	return ans;
}

Compilation message (stderr)

prize.cpp: In function 'int find_best(int)':
prize.cpp:46:6: warning: unused variable 'jumped' [-Wunused-variable]
   46 |  int jumped = 0, addPos = 0;
      |      ^~~~~~
prize.cpp:46:18: warning: unused variable 'addPos' [-Wunused-variable]
   46 |  int jumped = 0, addPos = 0;
      |                  ^~~~~~

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