| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1333010 | kawhiet | 선물상자 (IOI15_boxes) | C++20 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
#include "boxes.h"
using namespace std;
long long delivery(int n, int k, int len, int p[]) {
vector<int> pos;
for (int i = 0; i < n; i++) {
if (p[i] >= 0) {
pos.push_back(p[i]);
}
}
n = pos.size();
deque<int> l, r;
for (int i = 0; i < n; i++) {
if (pos[i] <= len / 2) {
l.push_back(pos[i]);
} else {
r.push_back(pos[i]);
}
}
ranges::reverse(r);
long long ans = 0;
for (int i = k - 1; i < l.size(); i += k) {
ans += l[i] * 2;
}
for (int i = k - 1; i < r.size(); i += k) {
ans += (len - r[i]) * 2;
}
while (l.size() >= k) {
for (int i = 0; i < k; i++) {
l.pop_front();
}
}
while (r.size() >= k) {
for (int i = 0; i < k; i++) {
r.pop_front();
}
}
if (!l.empty() && !r.empty() && l.size() + r.size() <= k) {
ans += len;
} else {
if (!l.empty()) ans += l.back() * 2;
if (!r.empty()) ans += (len - r.back()) * 2;
}
return ans;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, k, l;
cin >> n >> k >> l;
int pos[n];
for (int i = 0; i < n; i++) {
cin >> pos[i];
}
cout << delivery(n, k, l, pos) << '\n';
return 0;
}