| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1333160 | kawhiet | 선물상자 (IOI15_boxes) | C++20 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
using namespace std;
#define int long long
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);
for (int i = 0; i < k; 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;
}