# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
29509 | inqr | 쌀 창고 (IOI11_ricehub) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "ricehub.h"
#include <bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define rt insert
#define st first
#define nd second
using namespace std;
vector < int > presum;
vector < long long > presum2;
int fienum,ans;
long long b;
long long sum(int l,int r){
if(l==0)return presum[r];
else return presum[r]-presum[l-1];
}
long long comp(int l,int r,int X[]){
long long m=(l+r)/2;
long long lc=m-l,rc=r-m;
return lc*X[m]-sum(l,m-1)+sum(m+1,r)-rc*X[m];
}
void solve(int s,int X[]){
int l=s,r=fienum-1;
while(l<=r){
int m=(l+r)/2;
//printf("s=%d l=%d r=%d m=%d comp=%d %d\n",s,l,r,m,comp(s,m,X).st,comp(s,m,X).nd);
if(comp(s,m,X)<=b){
l=m+1;
ans=max(ans,m-s+1);
}
else r=m-1;
}
}
long long sum2(int l,int r){
if(l==0)return presum2[r];
else return presum2[r]-presum2[l-1];
}
long long comp2(int l,int r,int X[]){
long long m=(l+r)/2;
long long lc=m-l,rc=r-m;
return lc*X[m]-sum2(l,m-1)+sum2(m+1,r)-rc*X[m];
}
void solve2(int s,int X[]){
int l=s,r=fienum-1;
while(l<=r){
int m=(l+r)/2;
//printf("s=%d l=%d r=%d m=%d comp=%d %d\n",s,l,r,m,comp(s,m,X).st,comp(s,m,X).nd);
if(comp2(s,m,X)<=b){
l=m+1;
ans=max(ans,m-s+1);
}
else r=m-1;
}
}
int besthub(int R, int L, int X[], long long B)
{
fienum=R;b=B;
presum.resize(R);
presum[0]=presum2=X[0];
for(int i=1;i<R;i++){presum[i]=presum[i-1]+X[i];presum2[i]=presum2[i-1]+X[i];}
if(R<=5000)for(int i=0;i<R;i++)solve(i,X);
else for(int i=0;i<R;i++)solve2(i,X);
return ans;
}