Submission #593675

#	Time	Username	Problem	Language	Result	Execution time	Memory
593675		davi_bart	Boxes with souvenirs (IOI15_boxes)	C++17	50 / 100	2078 ms	49852 KiB

boxes

#pragma GCC optimize("O3")
#include <bits/stdc++.h>

#include "boxes.h"
#define ll long long
#define fi first
#define se second
#define pb push_back
#define int ll
using namespace std;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
vector<int> v;
int N, K, L;
int dist(int a, int b) {
    if (a > b) swap(a, b);
    return min(b - a, a + L - b);
}
long long delivery(signed n, signed k, signed l, signed p[]) {
    N = n;
    K = k;
    L = l;
    for (int i = 0; i < N; i++) v.pb(p[i]);
    vector<int> ans;
    vector<int> pref = {0};
    for (int i = 1; i < N; i++) {
        pref.pb(pref.back() + dist(v[i], v[i - 1]));
    }
    for (int i = 0; i <= N; i++) {
        int tot = 0;
        for (int j = i - 1; j >= 0; j -= K) {
            tot += dist(0, v[j]);
            tot += pref[j] - pref[max((int)0, j - K + 1)];
            // for (int h = j - 1; h >= max((int)0, j - K + 1); h--) {
            //     tot += dist(v[h + 1], v[h]);
            // }
            int o = v[0];
            if (j - K + 1 >= 0) o = v[j - K + 1];
            tot += dist(0, o);
        }
        ans.pb(tot);
        // cout << tot << " ";
    }
    // cout << endl;
    reverse(v.begin(), v.end());
    vector<int> ans1 = {0};
    vector<int> pref1 = {0};
    for (int i = 1; i < N; i++) {
        pref1.pb(pref1.back() + dist(v[i], v[i - 1]));
    }
    for (int i = 1; i <= N; i++) {
        int tot = 0;
        int j = i - 1;
        tot += dist(0, v[j]);
        tot += pref1[j] - pref1[max((int)0, j - K + 1)];
        // for (int h = j - 1; h >= max((int)0, j - K + 1); h--) {
        //     tot += dist(v[h + 1], v[h]);
        // }
        int o = v[0];
        if (j - K + 1 >= 0) o = v[j - K + 1];
        tot += dist(0, o);
        if (j - K >= 0) {
            tot += ans1[j - K + 1];
        }
        // for (int j = i - 1; j >= 0; j -= K) {
        //     tot += dist(0, v[j]);
        //     tot += pref1[j] - pref1[max((int)0, j - K + 1)];
        //     // for (int h = j - 1; h >= max((int)0, j - K + 1); h--) {
        //     //     tot += dist(v[h + 1], v[h]);
        //     // }
        //     int o = v[0];
        //     if (j - K + 1 >= 0) o = v[j - K + 1];
        //     tot += dist(0, o);
        // }
        ans1.pb(tot);
        // cout << tot << " ";
    }
    // cout << endl;
    int mi = 1e18;
    for (int i = 0; i <= N; i++) {
        // cout << ans[i] << " " << ans1[N - i] << endl;
        mi = min(mi, ans[i] + ans1[N - i]);
    }
    return mi;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 15 / 15
• Subtask 4: 15 / 15
• Subtask 5: 0 / 20
• Subtask 6: 0 / 30