Submission #715575

#TimeUsernameProblemLanguageResultExecution timeMemory
715575cbupRice Hub (IOI11_ricehub)C++14
68 / 100
16 ms1492 KiB
#include<bits/stdc++.h>
using namespace std;

long long findcost(int st,int en,int mid,vector<long long> &prefixSum,int *x){
    return (mid-st)*x[mid]*1LL-(prefixSum[mid]-prefixSum[st]) + (prefixSum[en+1]-prefixSum[mid+1])-(en-mid)*x[mid]*1LL;
}

long long besthub(int n,int l,int x[],long long b){
    long long cost,ans=1;
    int mid,st,en,m,check = 0;
    vector<long long> prefixSum(n+1,0);

    st = en = 0;
    for(int i=1;i<=n;i++)prefixSum[i] = prefixSum[i-1]+x[i-1];
    //for(int i=1;i<=n;i++)cout<<prefixSum[i]<<" "; cout<<endl;
    for(int i=0;i<n;i++){
        en = i;
        mid = (st+en)/2;
        cost = findcost(st,en,mid,prefixSum,x);

        if(cost <= b){
            ans = max(ans,en-st+1LL);
        }
        else{
            while(findcost(st,en,mid,prefixSum,x) > b){
                st++;
                mid = (st+en)/2;
            }
        }
        //cout<<st<<" "<<en<<" "<<cost<<endl;
    }

    return ans;
}

Compilation message (stderr)

ricehub.cpp: In function 'long long int besthub(int, int, int*, long long int)':
ricehub.cpp:10:19: warning: unused variable 'm' [-Wunused-variable]
   10 |     int mid,st,en,m,check = 0;
      |                   ^
ricehub.cpp:10:21: warning: unused variable 'check' [-Wunused-variable]
   10 |     int mid,st,en,m,check = 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...