제출 #528907

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
528907		d4xn	커다란 상품 (IOI17_prize)	C++17	20 / 100	84 ms	1412 KiB

prize

#pragma GCC optimize ("O3")
#include "prize.h"
#include <vector>
#include <map>
using namespace std;
 
map<int, vector<int>> dp;
 
int search(int l, int r) {
	if (l > r) return -1;
	if (l == r) {
		vector<int> ans;
		if (dp.find(l) != dp.end()) ans = dp[l];
		else {
			ans = ask(l);
			dp[l] = ans;
		}
 
		if (ans[0] == 0 && ans[1] == 0) return l;
		return -1;
	}
	
	int m = (l+r)/2;
	vector<int> ans;
	if (dp.find(m) != dp.end()) ans = dp[m];
	else {
		ans = ask(m);
		dp[m] = ans;
	}
	
	if (ans[0] == 0 && ans[1] == 0) {
		return m;
	}
	if (ans[0] == 0) {
		return search(m+1, r);
	}
	if (ans[1] == 0) {
		return search(l, m-1);
	}
	int ans1, ans2;
	ans1 = search(l, m-1);
  	if (ans1 != -1) return ans1;
  	ans2 = search(m+1, r);	
  	return ans2;
}
 
int find_best(int n) {
	return search(0, n-1);
}

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