제출 #1333162

#제출 시각아이디문제언어결과실행 시간메모리
1333162kawhiet선물상자 (IOI15_boxes)C++20
50 / 100
110 ms20140 KiB
#include <bits/stdc++.h>
using namespace std;

long long get(vector<int> a, int k) {
    if (a.empty()) return 0;
    int n = a.size();
    multiset<long long> s;
    vector<long long> dp(n);
    for (int i = 0; i < min(n, k); i++) {
        dp[i] = a[i] * 2;
        s.insert(dp[i]);
    }
    for (int i = k; i < n; i++) {
        dp[i] = *s.begin() + a[i] * 2;
        s.erase(s.find(dp[i - k]));
        s.insert(dp[i]);
    }
    return dp[n - 1];
}

long long delivery(int n, int k, int len, int p[]) {
    vector<int> pos(n);
    for (int i = 0; i < n; i++) {
        pos[i] = p[i];
    }
    n = pos.size();
    vector<int> l, r;
    for (int i = 0; i < n; i++) {
        if (pos[i] <= len / 2) {
            l.push_back(pos[i]);
        } else {
            r.push_back(len - pos[i]);
        }
    }
    ranges::reverse(r);
    long long ans = get(l, k) + get(r, k);
    if (l.size() % k + r.size() % k <= k) {
        int rem = k - l.size() % k - r.size() % k;
        while (l.size() % k > 0) l.pop_back();
        while (r.size() % k > 0) r.pop_back();
        for (int i = 0; i < rem; i++) {
            if (l.empty()) {
                r.pop_back();
            } else if (r.empty()) {
                l.pop_back();
            } else {
                if (l.back() > r.back()) {
                    l.pop_back();
                } else {
                    r.pop_back();
                }
            }
        }
        ans = min(ans, get(l, k) + get(r, k) + len);
    }
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...