Submission #637573

#	Time	Username	Problem	Language	Result	Execution time	Memory
637573		boris_mihov	The Big Prize (IOI17_prize)	C++17	20 / 100	88 ms	1872 KiB

prize

#include "prize.h"
#include <algorithm>
#include <iostream>
#include <cassert>
#include <numeric>
#include <vector>

typedef long long llong;
const int MAXN = 200000 + 10;
const int INF  = 1e9;

int ans, n;
std::vector <int> res;
std::pair <int,int> asked[MAXN];
inline std::pair <int,int> query(int pos)
{
    if (asked[pos].first != -1) return asked[pos];
    res = ask(pos - 1);
    if (res[0] == 0 && res[1] == 0) ans = pos;
    return asked[pos] = {res[0], res[1]};
}

int leftCnt;
int rightCnt;
int cntBig;
int binaryCnt;

int binarySearchLeft(int from, int to)
{
    int l = from, r = to, mid;
    while (l < r - 1)
    {
        int mid = (l + r) / 2;
        auto [left, right] = query(mid);
        if (ans != 0) return -1;
        if (left + right != cntBig)
        {
            r = mid;
            continue;
        }

        if (left != leftCnt)
        {
            r = mid;
            continue;
        }

        l = mid;
    }

    assert(r != n+1);
    query(r);
    leftCnt++;
    return r;
}

int binarySearchRight(int from, int to)
{
    int l = from, r = to, mid;
    while (l < r - 1)
    {
        int mid = (l + r) / 2;
        auto [left, right] = query(mid);
        if (ans != 0) return -1;
        if (left + right != cntBig)
        {
            l = mid;
            continue;
        }

        if (right != rightCnt)
        {
            l = mid;
            continue;
        }

        r = mid;
    }

    query(l);
    rightCnt++;
    return l;
}

int find_best(int N) 
{
    n = N;
	srand(6686889);
    for (int i = 1 ; i <= n ; ++i)
    {
        asked[i].first = -1;
        asked[i].second = -1;
    }
    for (int i = 1 ; i <= 10 ; ++i)
    {
        int rndPos = rand()%n + 1;
        cntBig = std::max(cntBig, query(rndPos).first + query(rndPos).second);
    }

    int l = 0, r = n + 1;
    while (ans == 0 && l < r - 1)
    {
        l = binarySearchLeft(l, r);
        // if (rand()%2 == 0) l = binarySearchLeft(l, r);
        // else               r = binarySearchRight(l, r);
    }

    if (ans == 0 && l != 0) query(l);
    if (ans == 0 && r != n + 1) query(r);
    assert(1 <= ans && ans <= n && asked[ans].first == 0 && asked[ans].second == 0);
    return ans - 1;
}

Compilation message (stderr)

prize.cpp: In function 'int binarySearchLeft(int, int)':
prize.cpp:30:27: warning: unused variable 'mid' [-Wunused-variable]
   30 |     int l = from, r = to, mid;
      |                           ^~~
prize.cpp: In function 'int binarySearchRight(int, int)':
prize.cpp:59:27: warning: unused variable 'mid' [-Wunused-variable]
   59 |     int l = from, r = to, mid;
      |                           ^~~

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