#include "ricehub.h"
using ll = long long;
#define MIN(a, b) (((a) < (b)) ? (a) : (b))
#define MAX(a, b) (((a) < (b)) ? (b) : (a))
int const nmax = 100000;
int n, lim;
ll v[1 + nmax], sum[1 + nmax];
///returns position of the last farm with coordinate lower (or equal) than target
int lowerthan(int target){
int x = 0;
for(int jump = (1 << 20); 0 < jump; jump /= 2)
if(x + jump <= n && v[x + jump] <= target)
x += jump;
return x;
}
ll tryit(int pos, ll dist, ll budget){
int pos2 = lowerthan(v[pos] + dist), pos3 = lowerthan(v[pos] - dist - 1) + 1;
ll result = ((sum[pos2] - sum[pos - 1]) - (pos2 - pos + 1) * v[pos]);
ll result2 = (pos - pos3 + 1) * v[pos] - (sum[pos] - sum[pos3 - 1]);
return result + result2 <= budget;
}
ll solve(int pos, ll budget){
int dist = 0;
for(int jump = (1 << 30); 0 < jump; jump /= 2){
if(dist + jump <= lim && tryit(pos, dist + jump, budget))
dist += jump;
}
dist++;
int pos2 = lowerthan(v[pos] + dist), pos3 = lowerthan(v[pos] - dist - 1) + 1;
ll result = ((sum[pos2] - sum[pos - 1]) - (pos2 - pos + 1) * v[pos]);
ll result2 = (pos - pos3 + 1) * v[pos] - (sum[pos] - sum[pos3 - 1]);
result = result + result2;
if(budget <= result)
return (pos2 - pos3 + 1);
else
return (pos2 - pos3 + 1) - (result - budget + dist - 1) / dist;
}
int besthub(int R, int L, int X[], long long B)
{
lim = L;
n = R;
for(int i = 1;i <= n; i++)
v[i] = X[i - 1];
for(int i = 1;i <= n; i++)
sum[i] = sum[i - 1] + v[i];
int result = 0;
for(int i = 1;i <= n; i++) {
int result2 = solve(i, B);
result = MAX(result, result2);
}
return result;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
256 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
256 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
43 ms |
760 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |