#include <bits/stdc++.h>
using namespace std;
// types - only for stuff used a lot
using ll = long long;
#define vv vector
#define Pp pair
// IO
#define get(x) scanf("%d",&x)
#define getl(x) scanf("%lld",&x);
// Operations
#define pb push_back
#define pob pop_back
#define sz(a) int(a.size())
#define re(a,b) a=max(a,b) // relax
#define ri(a,b) a=min(a,b) // relaxi
// Debugging
#ifndef LOCAL
#define cerr if (0) cerr
#else
#define cerr cout
#endif
#define print(arr,n) {for (int _ = 0; _ < n; _++) cerr<<arr[_]<<" "; cerr << endl; }
#define printv(vec) {for (int _ : vec) cerr<<_<<" "; cerr<<endl;}
const int mod = 1e9+7, oo = 1e9;
const ll loo = 1e18;
// Functions
ll modpow(ll a, ll b) {
ll ans = 1; // faster modpow than recursive
for (; b; b/=2,a=a*a%mod)
if (b&1) ans = (ans*a)%mod;
return ans;
}
ll gcd(ll a, ll b) {
while (a) b%=a,swap(a,b);
return b;
}
#define bitcount __builtin_popcountll
#define f(i,a,b) for (int i = a; i < b; i++)
#define fr(i,a,b) for (int i = b-1; i >= a; i--)
/*
ALRIGHT HACKERS, THIS IS WHERE THE ACTUAL CODE BEGINS
*/
const bool DEBUG = 1;
using pii = pair<int,int>;
int best_hub(int N, int L, int rem[], int B) {
ll C[N];
f(i,0,N) C[i] = i?rem[i]+C[i-1]:rem[i];
vector<int> X(N);
f(i,0,N) X[i] = rem[i];
auto sum = [&](int i, int j) {
if (i>j) return 0LL;
return C[j]-(i?C[i-1]:0);
};
auto fu = [&](int i, int l) {
// to the left
int x = lower_bound(begin(X),end(X),X[i]-l)-begin(X);
int y = upper_bound(begin(X),end(X),X[i]+l)-begin(X)-1;
return pii(1LL*(i-x)*X[i]-sum(x,i-1)+
sum(i+1,y)-1LL*(y-i)*X[i],y-x+1);
};
int ans = 0;
f(i,0,N) {
int lo = 0, hi = L;
while (lo <= hi) {
int mid = lo+hi>>1;
if (fu(i,mid).first <= B)
lo = mid+1;
else hi = mid-1;
}
re(ans,fu(i,hi).second);
}
return ans;
}
Compilation message
ricehub.cpp: In function 'int best_hub(int, int, int*, int)':
ricehub.cpp:82:16: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
int mid = lo+hi>>1;
~~^~~
/tmp/ccXMwhUx.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
