Submission #316216

#	Time	Username	Problem	Language	Result	Execution time	Memory
316216		nikatamliani	The Big Prize (IOI17_prize)	C++14	91.25 / 100	70 ms	5368 KiB

prize

#include <bits/stdc++.h>
using namespace std;
#include "prize.h"
vector < vector < int > > memo;
vector < int > memo_ask(int x) {
    if(!memo[x].empty()) return memo[x];
    return memo[x] = ask(x);
}
int find_best(int n) {
    const int BLOCK_SIZE = 250;
    int next = -1, at_most = min(500, n - 1);
    vector < int > bad(2); 
    memo.resize(n);
    for(int i = 0; i <= at_most; ++i) {
        vector < int > cur = memo_ask(i);
        if(bad[0] + bad[1] < cur[0] + cur[1]) bad = cur;
    }
    for(int i = 0; i < n; i = next + 1) {
        int l = i, r = n - 1; next = -1;
        vector < int > t = memo_ask(i);
        if(t[0] + t[1] != bad[0] + bad[1]) {
            next = i;
            if(t[0] == 0 && t[1] == 0) return i; 
            continue;
        } 
        int maybe = i + BLOCK_SIZE;
        if(maybe < n && memo_ask(maybe) == t) {
            next = maybe;
            continue;
        }
        if(maybe < n) r = maybe - 1;
        while(r >= l) {
            int m = l + r >> 1;
            vector < int > x = memo_ask(m);
            if(x == t) {
                next = m;
                l = m + 1;
            } else {
                r = m - 1; 
            }
        }
        assert(next >= i);
    }
}

Compilation message (stderr)

prize.cpp: In function 'int find_best(int)':
prize.cpp:33:23: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   33 |             int m = l + r >> 1;
      |                     ~~^~~
prize.cpp:12:25: warning: control reaches end of non-void function [-Wreturn-type]
   12 |     vector < int > bad(2);
      |                         ^

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