#include "ricehub.h"
#include <bits/stdc++.h>
using namespace std;
# define int long long
long long ps[100100];
int besthub(int n, int mx, int a[], long long b)
{
for ( int i = 0; i < n; i++ ) {
ps[i] = a[i];
if ( i ) {
ps[i] += ps[i - 1];
}
}
int ans = 1;
for ( int i = 0; i < n; i++ ) {
int x = a[i];
// cout << x << '\n';
// for ( auto j: vc ) {
// cout << j << ' ';
// }
// cout << "\n";
int l = i + 1, r = n - 1, mid;
while ( l <= r ) {
mid = (l + r) / 2;
long long sum = ps[mid] - ps[i];
sum -= (mid - i) * x;
long long left = b - sum;
if ( left < 0 ) {
r = mid - 1;
}
else {
// cout << mid << '\n';
ans = max( ans, mid - i + 1 );
int ll = 0, rr = i - 1, mid1, ind = i;
while ( ll <= rr ) {
mid1 = (ll + rr) / 2;
long long sum1 = ps[i - 1];
if ( mid1 ) {
sum1 -= ps[mid1 - 1];
}
sum1 = (i - mid1) * x - sum1;
if ( sum1 > left ) {
ll = mid1 + 1;
}
else {
rr = mid1 - 1;
ind = mid1;
// ans = max( ans, (mid - i) + (i - ind) + 1 );
}
}
/*
auto it = upper_bound( vc.begin(), vc.end(), left );
// it--;
if ( it != vc.begin() ) {
it--;
}
int ind = it - vc.begin();*/
// cout << a[j] << ' ' << pr << ' ' << ind << '\n';
ans = max( ans, (mid - i) + (i - ind) + 1 );
l = mid + 1;
}
}
/*
long long pr = 0;
for ( int j = i + 1; j < n; j++ ) {
pr += a[j] - x;
int left = b - pr;
// cout << a[j] << ' ' << left << '\n';
if ( left < 0 ) {
continue;
}
ans = max( ans, j - i + 1 );
auto it = upper_bound( vc.begin(), vc.end(), left );
// it--;
if ( it != vc.begin() ) {
it--;
}
int ind = it - vc.begin();
// cout << a[j] << ' ' << pr << ' ' << ind << '\n';
ans = max( ans, j - i + ind + 1 );
}*/
}
return ans;
}
Compilation message
/tmp/ccntu8IZ.o: In function `main':
grader.cpp:(.text.startup+0x92): undefined reference to `besthub(int, int, int*, long long)'
collect2: error: ld returned 1 exit status
