# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
950456 | 2024-03-20T10:24:18 Z | irmuun | Rice Hub (IOI11_ricehub) | C++17 | 0 ms | 0 KB |
#include<bits/stdc++.h> using namespace std; #define ll long long #define pb push_back #define ff first #define ss second #define all(s) s.begin(),s.end() #define rall(s) s.rbegin(),s.rend() int besthub(int R,int L,int X[],ll B){ ll sum[R+1]; sum[0]=0; for(int i=0;i<R;i++){ sum[i+1]=sum[i]+X[i]; } int ans=0; for(int i=1;i<=R;i++){ int l=0,r=n; while(l<r){ int mid=(l+r+1)/2; if(i-mid<=0||i+mid>R){ r=mid-1; continue; } ll cost=1ll*X[i-1]*l-(sum[i-1]-sum[i-mid-1])+(sum[i+mid]-sum[i])-1ll*X[i-1]*l; if(cost<=B){ l=mid; } else{ r=mid-1; } } ans=max(ans,l*2+1); } for(int i=1;i<R;i++){ if(X[i]-X[i-1]>B) continue; int l=0,r=n; while(l<r){ int mid=(l+r+1)/2; if(i-mid<=0||i+1+mid>R){ r=mid-1; continue; } ll cost=1ll*X[i-1]*l-(sum[i-1]-sum[i-mid-1])+(sum[i+mid+1]-sum[i+1])-1ll*X[i]*l; if(cost<=B){ l=mid; } else{ r=mid-1; } } ans=max(ans,l*2+2); } return ans; }