제출 #1333165

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

vector<long long> get(vector<int> a, int k) {
    if (a.empty()) return {};
    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;
}

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);
    auto x = get(l, k);
    auto y = get(r, k);
    long long ans = 0;
    if (!x.empty()) ans += x.back();
    if (!y.empty()) ans += y.back();
    long long mn = 1e18;
    for (int i = 0; i <= k; i++) {
        if (l.size() < i) continue;
        if (r.size() < k - i) continue;
        long long lhs = (l.size() == i ? 0 : x[x.size() - 1 - i]);
        long long rhs = (r.size() == k - i ? 0 : y[y.size() - 1 - k + i]);
        mn = min(mn, lhs + rhs + len);
    }
    ans = min(ans, mn);
    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...