#include "ricehub.h"
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define N 100005
vector<int> ps(N, 0);
bool ok(int L, int R, int x) {
int mid = (L + R) >> 1; // median
// brute force
int cost = v[mid] * (mid - L + 1) - (ps[mid] - ps[L - 1]);
// cout << L << " " << R << " " << mid << " " << cost;
cost += (ps[R] - ps[mid]) - v[mid] * (R - mid);
// cout << " " << cost << "\n";
return (cost <= b);
}
bool check(int x) {
// x consecutive hubs
for (int i = 1; i <= r - x + 1; i++) {
if (ok(i, i + x - 1, x)) return true;
}
return false;
}
int besthub(int r, int l, int v[], long long b) {
int lo = 1, hi = r, ans;
while (lo <= hi) {
int m = (lo + hi) >> 1;
// cout << m << " " << check(m) << "\n";
if (check(m)) lo = m + 1, ans = m;
else hi = m - 1;
}
return ans;
}
Compilation message
ricehub.cpp: In function 'bool ok(long long int, long long int, long long int)':
ricehub.cpp:10:16: error: 'v' was not declared in this scope
10 | int cost = v[mid] * (mid - L + 1) - (ps[mid] - ps[L - 1]);
| ^
ricehub.cpp:14:21: error: 'b' was not declared in this scope
14 | return (cost <= b);
| ^
ricehub.cpp: In function 'bool check(long long int)':
ricehub.cpp:19:26: error: 'r' was not declared in this scope
19 | for (int i = 1; i <= r - x + 1; i++) {
| ^
