제출 #641356

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
641356		ggoh	커다란 상품 (IOI17_prize)	C++14	20 / 100	120 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;i<=n;i++)
  {
    if(1+i+(i*i+1)<=n)m=max(m,1+i);
    else i=n+1;
  }
  for(int i=2;i<=n;i++)
  {
    for(int j=i*i+1;j<=n;j++)
    {
      if(1+i+j+(j*j+1)<=n)m=max(m,1+i+j);
      else j=n+1,i=n+1;
    }
  }
  for(int i=2;i<=n;i++)
  {
    for(int j=i*i+1;j<=n;j++)
    {
      for(int k=j*j+1;k<=n;k++)
      {
        if(1+i+j+k+(k*k+1)<=n)m=max(m,1+i+j+k);
        else k=n+1,j=n+1,i=n+1;
      }
    }
  }
  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: 0 / 80