Submission #486152

# Submission time Handle Problem Language Result Execution time Memory
486152 2021-11-10T15:56:38 Z Jomnoi Rice Hub (IOI11_ricehub) C++17
Compilation error
0 ms 0 KB
#include <bits/stdc++.h>
#include <ricehub.h>
#define DEBUG 0
using namespace std;

int cost(int R, int X[], int mid, long long B) {
    long long closest, sum = 0;
    int l, r, ct = 0, ok;
    r = lower_bound(X, X + R, mid) - X;
    l = r - 1;
    while(1) {
        closest = 1e18;
        if(l >= 0 && closest > abs(mid - X[l])) {
            closest = abs(mid - X[l]);
            ok = 0;
        }
        if(r < R && closest > abs(X[r] - mid)) {
            closest = abs(X[r] - mid);
            ok = 1;
        }
        if(sum + closest > B) {
            return ct;
        }
        ct++;
        sum += closest;
        if(!ok) {
            l--;
        }
        else {
            r++;
        }
    }
    return R;
}

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;
}
 
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int r, l, b;
    cin >> r >> l >> b;
    int x[l];
    for(int i = 0; i < r; i++) {
        cin >> x[i];
    }
    cout << besthub(r, l, x, b);
    return 0;
}
 
/*
5 20 6
1 2 10 12 14
*/

Compilation message

ricehub.cpp: In function 'int cost(int, int*, int, long long int)':
ricehub.cpp:26:9: warning: 'ok' may be used uninitialized in this function [-Wmaybe-uninitialized]
   26 |         if(!ok) {
      |         ^~
/usr/bin/ld: /tmp/ccdnYMN3.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccUwJR11.o:ricehub.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status