Submission #285218

#TimeUsernameProblemLanguageResultExecution timeMemory
285218ec1117Rice Hub (IOI11_ricehub)C++17
42 / 100
21 ms1536 KiB
#include <bits/stdc++.h>
#include "ricehub.h"

using namespace std;

typedef long long ll;
typedef long double ld;
typedef complex<ld> cd;

typedef pair<int, int> pi;
typedef pair<ll,ll> pl;
typedef pair<ld,ld> pd;

typedef vector<int> vi;
typedef vector<ld> vd;
typedef vector<ll> vl;
typedef vector<pi> vpi;
typedef vector<pl> vpl;
typedef vector<cd> vcd;

#define FOR(i, a, b) for (int i=a; i<(b); i++)
#define F0R(i, a) for (int i=0; i<(a); i++)
#define FORd(i,a,b) for (int i = (b)-1; i >= a; i--)
#define F0Rd(i,a) for (int i = (a)-1; i >= 0; i--)
#define trav(a,x) for (auto& a : x)
#define uid(a, b) uniform_int_distribution<int>(a, b)(rng)

#define sz(x) (int)(x).size()
#define mp make_pair
#define pb push_back
#define F first
#define S second
#define lb lower_bound
#define ub upper_bound
#define all(x) x.begin(), x.end()
#define ins insert

const int MOD = 1000000007;
const char nl = '\n';
const int MX = 100001; //check the limits, dummy

int besthub(int r,int l, int x[],ll b) {
    int lhub=0;
    int p=0;
    int rhub=0;
    ll rice=0;//riceused
    ll y[r];
    for(int i=0;i<r;i++)y[i]=x[i];
    sort(y,y+r);
    int mx=1;
    while(rhub<r-1 && rice+y[rhub+1]-y[p]<=b){
        rhub++;
        rice+=y[rhub]-y[p];
    }
    mx=max(mx, rhub-lhub+1);
    while(p<r-1 && rhub<r){
        rice+=-(y[p+1]-y[p])*((rhub-p-1)-(p-lhub));
        p++;
        while(rice>b){//more hubs to the left than right
            rice-=y[p]-y[lhub];//remove lhub
            lhub++;
        }
        while(rhub<r-1 && p<r-1 && rice-(y[p]-y[lhub])+(y[rhub+1]-y[p])<=b){//do i add rhub+1?
            rice-=y[p]-y[lhub];
            lhub++;
            while(rhub<r-1 && rice+y[rhub+1]-y[p]<=b){
                rhub++;
                rice+=y[rhub]-y[p];
            }
            if(lhub>p)break;
        }
        while(lhub!=0 && rice+(y[p]-y[lhub-1])<=b){
            lhub--;
            rice+=y[p]-y[lhub];
        }
        mx=max(mx,rhub-lhub+1);
    }
    return mx;
}
//
//int main(){
//    int x[]={1,3,7};
//    cout << besthub(3,10,x,1)<<nl;
//    cout << besthub(3,10,x,2)<<nl;
//    cout << besthub(3,10,x,4)<<nl;
//    cout << besthub(3,10,x,6)<<nl;
//    return 0;
//}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...