This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include <ricehub.h>
#define DEBUG 0
using namespace std;
int cost(int n, int X[], int mid, long long B) {
deque <int> L, R;
for(int i = 0; i < n; i++) {
if(X[i] <= mid) {
L.push_front(mid - X[i]);
}
else {
R.push_back(X[i] - mid);
}
}
int i = 0, j = 0;
long long sum = 0;
while(i < L.size() && j < R.size()) {
if(L[i] <= R[j]) {
sum += L[i++];
}
else {
sum += R[j++];
}
if(sum > B) {
return i + j - 1;
}
}
while(i < L.size()) {
sum += L[i++];
if(sum > B) {
return i + j - 1;
}
}
while(j < R.size()) {
sum += R[j++];
if(sum > B) {
return i + j - 1;
}
}
return n;
}
int besthub(int R, int L, int X[], long long B) {
int l = 1, r = L, ans = 0, mid, cost1;
int l2, r2, mid2, pos2, cost2;
while(l <= r) {
mid = (l + r) / 2;
cost1 = cost(R, X, mid, B);
l2 = l + 1, r2 = r, pos2 = r;
while(l2 <= r2) {
mid2 = (l2 + r2) / 2;
if(cost1 == cost(R, X, mid2, B)) {
l2 = mid2 + 1;
}
else {
r2 = mid2 - 1;
pos2 = min(pos2, mid2);
}
}
cost2 = cost(R, X, pos2, B);
if(cost1 > cost2) {
r = mid - 1;
ans = max(ans, cost1);
}
else {
l = mid + 1;
ans = max(ans, cost2);
}
}
return ans;
}
Compilation message (stderr)
ricehub.cpp: In function 'int cost(int, int*, int, long long int)':
ricehub.cpp:18:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
18 | while(i < L.size() && j < R.size()) {
| ~~^~~~~~~~~~
ricehub.cpp:18:29: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
18 | while(i < L.size() && j < R.size()) {
| ~~^~~~~~~~~~
ricehub.cpp:29:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
29 | while(i < L.size()) {
| ~~^~~~~~~~~~
ricehub.cpp:35:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
35 | while(j < R.size()) {
| ~~^~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |