Submission #715579

#	Time	Username	Problem	Language	Result	Execution time	Memory
715579		vjudge1	Rice Hub (IOI11_ricehub)	C++14	100 / 100	12 ms	1492 KiB

ricehub

#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,len,low=0,high=n-1,m,check = 0;
    vector<long long> prefixSum(n+1,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;
    while(low<=high){
        check = 0;
        len = (low+high)/2;
        for(st=0;st<n-len;st++){
            en = st+len;
            mid = (st+en)/2;
            cost = findcost(st,en,mid,prefixSum,x);

            if(cost <= b){
                ans = max(ans,en-st+1LL);
                check = 1;
                break;
            }
            //cout<<st<<" "<<en<<" "<<cost<<endl;
        }

        if(!check)high = len-1;
        else low = len+1;
    }

    return ans;
}

Compilation message (stderr)

ricehub.cpp: In function 'long long int besthub(int, int, int*, long long int)':
ricehub.cpp:10:38: warning: unused variable 'm' [-Wunused-variable]
   10 |     int mid,st,en,len,low=0,high=n-1,m,check = 0;
      |                                      ^

• Subtask 1: 17 / 17
• Subtask 2: 25 / 25
• Subtask 3: 26 / 26
• Subtask 4: 32 / 32