# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
311972 | jainbot27 | Rice Hub (IOI11_ricehub) | C++17 | 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>
using namespace std;
#define f first
#define s second
#define pb push_back
#define ar array
#define all(x) x.begin(), x.end()
#define siz(x) (int)x.size()
#define FOR(x, y, z) for(int x = (y); x < (z); x++)
#define ROF(x, z, y) for(int x = (y-1); x >= (z); x--)
#define F0R(x, z) FOR(x, 0, z)
#define R0F(x, z) ROF(x, 0, z)
#define trav(x, y) for(auto&x:y)
using ll = long long;
using vi = vector<int>;
using vl = vector<long long>;
using pii = pair<int, int>;
using vpii = vector<pair<int, int>>;
template<class T> inline bool ckmin(T&a, T b) {return b < a ? a = b, 1 : 0;}
template<class T> inline bool ckmax(T&a, T b) {return b > a ? a = b, 1 : 0;}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const char nl = '\n';
const int mxN = 2e5 + 10;
const int MOD = 1e9 + 7;
const long long infLL = 1e18;
int besthub(int R, int L, int* X, int B){
#define int long long
ll p[R], sum;
F0R(i, R){
if(i) p[i] = p[i-1] + X[i];
else p[i] = X[i];
// cout << p[i] << ' ';
}
// cout << nl;
int i = 0, res = 0;
F0R(j, R){
sum = (((i+j)/2 - i + 1) - (j-(i+j)/2)) * X[(i+j)/2] - (p[(i+j)/2] - (i?p[i-1]:0)) + (p[j] - p[(i+j)/2]);
// cout << j << " " << i << " " << (j-i+1)*(X[(i+j)/2]) << " " << (p[(i+j)/2] - (i?p[i-1]:0)) << " " << (p[j] - p[(i+j)/2])<< endl;
// cout << i << ' ' << j << ' ' << sum << nl;
while(sum > B){
i++;
// sum = (j - i + 1) * X[(i+j)/2] - (p[(i+j)/2] - (i?p[i-1]:0)) + (p[j] - p[(i+j)/2]);
sum = (((i+j)/2 - i + 1) - (j-(i+j)/2)) * X[(i+j)/2] - (p[(i+j)/2] - (i?p[i-1]:0)) + (p[j] - p[(i+j)/2]);
// cout << j << " " << i << " " << sum << endl;
// cout << j << " " << i << " " << (j-i+1)*(X[(i+j)/2]) << " " << (p[(i+j)/2] - (i?p[i-1]:0)) << " " << (p[j] - p[(i+j)/2])<< endl;
// cout << i<< ' ' << j << ' ' << sum << nl;
}
ckmax(res, j - i + 1);
}
return res;
}
// int32_t main(){
// ios_base::sync_with_stdio(0); cin.tie(0);
// int X[] = {1, 2, 10, 12, 14};
// cout << besthub(5, 20, X, 6) << nl;
// return 0;
// }