Submission #827125

#	Time	Username	Problem	Language	Result	Execution time	Memory
827125		wortelworm	Rice Hub (IOI11_ricehub)	C++17	49 / 100	12 ms	1748 KiB

ricehub

#include "ricehub.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int besthub(int RiceCount, int Length, int RiceFields[], long long Budget)
{
    ll count_before = 0;
    ll count_after = 0;
    ll cost = 0;

    ll best_count = 1;

    for (ll i = 1; i < RiceCount; i++)
    {
        // set the hub at this field
        ll x = RiceFields[i];
        ll distance = RiceFields[i] - RiceFields[i-1];

        // account for the prev_index now also having distance
        count_before++;

        cost += distance * count_before;
        cost -= distance * count_after;

        if (count_after > 0) {
            count_after --;
        }

        auto add_after = [&](){
            while (cost <= Budget && i+count_after < RiceCount-1) {
                count_after++;
                cost += RiceFields[i+count_after] - x;
            }

            if (cost > Budget && count_after > 0) {
                cost -= RiceFields[i+count_after] - x;
                count_after--;
            }
        };


        add_after();
        while (
            count_before > 0
            && (
                cost > Budget
                || (
                    i+count_after < RiceCount-1 &&
                    (x - RiceFields[i-count_before]) > (RiceFields[i+count_after+1] - x)
                )
            )) {
    
            cost -= x - RiceFields[i-count_before];
            count_before--;

            add_after();
        }

        ll current_count = count_before + count_after + 1;
        if (current_count > best_count) {
            best_count = current_count;
        }
    }

    return best_count;
}

• Subtask 1: 17 / 17
• Subtask 2: 0 / 25
• Subtask 3: 0 / 26
• Subtask 4: 32 / 32