Submission #556301

#	Time	Username	Problem	Language	Result	Execution time	Memory
556301		Dan4Life	Rice Hub (IOI11_ricehub)	C++17	68 / 100	14 ms	1424 KiB

ricehub

#include "ricehub.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long

ll pref[100010];

ll sum(int i, int j){ return pref[j+1]-pref[i]; }

int besthub(int n, int m, int a[], ll k)
{
    /*
    for(int i = 0; i < n; i++){
        int j = i;
        while(i<n and a[i]==a[i+1]) i++;
        v.pb({a[j],j-i+1});
    }
    n = (int)v.size();*/
    pref[0] = 0;
    for(int i = 0; i < n; i++)
        pref[i+1] = pref[i]+(ll)a[i];//v[i].second;
    int l = 1, r = n;
    while(l<r){
        int mid = (l+r+1)/2; ll mn = (ll)4e18;
        for(int i = 0; i <= n-mid; i++){
            int j = i+mid-1; int med = (i+j)/2; int x = med;
            ll sum1 = (ll)a[med]*(med-i+1)-sum(i,med);
            ll sum2 = sum(med+1,j)-a[med]*(j-med);
            mn = min(mn, sum1+sum2);
        }
        if(mn<=k) l=mid;
        else r = mid-1;
    }
    return l;
}

Compilation message (stderr)

ricehub.cpp: In function 'int besthub(int, int, int*, long long int)':
ricehub.cpp:26:53: warning: unused variable 'x' [-Wunused-variable]
   26 |             int j = i+mid-1; int med = (i+j)/2; int x = med;
      |                                                     ^

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