# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
336111 | knightron0 | Rice Hub (IOI11_ricehub) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include "ricehub.h"
#define pb push_back
#define int long long
using namespace std;
const int MOD = 1e9 + 7;
const int MAXN = 1e5 + 5;
int a[MAXN], pref[MAXN];
int n;
bool check(long long x, long long b){
for(long long i= 1;i<=n-x+1;i++){
vector<long long> md;
if(x & 1){
md.pb((i + i +x-1)/2);
} else {
md.pb((i + i +x-1)/2);
md.pb(((i + i +x-1)/2)+1);
}
for(long long m: md){
long long sum1 = (long long)pref[m]-(long long)pref[i-1];
sum1 *= -1;
sum1 += (long long)(m-(i-1))*(long long)a[m];
long long sum2 = (long long)pref[i+x-1]-(long long)pref[m];
sum2 -= (long long)(i+x-1-m)*(long long)a[m];
long long sum = sum1+sum2;
if(sum <= b){
return true;
}
}
}
return false;
}
int besthub(int r, int l, int x[], long long b) {
n = r;
pref[0] = 0;
for(int i= 1;i<=n;i++){
a[i] = x[i-1];
pref[i] = pref[i-1] + a[i];
}
int lo = 1;
int hi = n;
while(lo <= hi){
int mid = lo+(hi-lo)/2;
if(check((long long) mid, b)){
lo = mid+1;
} else {
hi = mid-1;
}
}
if(!check(lo, b)){
return hi;
}
return lo;
}
// signed main() {
// ios_base::sync_with_stdio(false);
// cin.tie(NULL);
// #ifndef ONLINE_JUDGE
// freopen("input.txt", "r", stdin);
// #endif
// int m, k;
// cin>>n>>m>>k;
// pref[0] = 0;
// for(int i=1;i<=n;i++){
// cin>>a[i];
// pref[i] = pref[i-1] + a[i];
// }
// return 0;
// }