제출 #641361

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
641361		ggoh	커다란 상품 (IOI17_prize)	C++14	97.70 / 100	80 ms	1924 KiB

prize

#include "prize.h"
#include<bits/stdc++.h>
using namespace std;
#define sz(v) ((int)(v).size())
typedef long long lint;

int a[200000][2],n;
vector<int>ret;
int maxa;
void f(int p,int q,int left,int right)
{
  int h=(p+q)/2;
  if(a[h][0]==-1)
  {
    ret=ask(h);
    a[h][0]=ret[0];
    a[h][1]=ret[1];
  }
  int lh=h,rh=h;
  while(lh>p && a[lh][0]+a[lh][1]!=maxa)
  {
    lh--;
    if(a[lh][0]==-1)
    {
      ret=ask(lh);
      a[lh][0]=ret[0];
      a[lh][1]=ret[1];
    }
  }
  while(rh<q && a[rh][0]+a[rh][1]!=maxa)
  {
    rh++;
    if(a[rh][0]==-1)
    {
      ret=ask(rh);
      a[rh][0]=ret[0];
      a[rh][1]=ret[1];
    }
  }
  if(p!=lh)
  {
    if(a[lh][0]!=left)f(p,lh-1,left,a[lh][1]);
  }
  if(q!=rh)
  {
    if(a[rh][1]!=right)f(rh+1,q,a[rh][0],right);
  }
}
int find_best(int N) {
  n=N;
  memset(a,-1,sizeof(a));
  if(n<=5000)
  {
    for(int i=0;i<n;i++)
    {
      ret=ask(i);
      if(!ret[0] && !ret[1])return i;
    }
  }
  maxa=-1;
  int m=0;
  for(int i=2;1+i+(i*i+1)<=n;i++)
  {
    m=max(m,1+i);
  }
  for(int i=2;1+i+(i*i+1)<=n;i++)
  {
    for(int j=i*i+1;1+i+j+1ll*j*j+1<=n;j++)
    {
      m=max(m,1+i+j);
    }
  }
  for(int i=2;1+i+(i*i+1)<=n;i++)
  {
    for(int j=i*i+1;1+i+j+1ll*j*j+1<=n;j++)
    {
      for(int k=j*j+1;1+i+j+k+1ll*k*k+1<=n;k++)
      {
        m=max(m,1+i+j+k);
      }
    }
  }
  for(int i=0;i<m;i++)
  {
    if(a[i][0]==-1)
    {
      ret=ask(i);
      a[i][0]=ret[0];
      a[i][1]=ret[1];
    }
    if(!a[i][0] && !a[i][1])return i;
    maxa=max(maxa,a[i][0]+a[i][1]);
  }
  f(0,n-1,0,0);
  for(int i=0;i<n;i++)if(!a[i][0] && !a[i][1])return i;
  return 0;//unreachable
}

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