| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 968595 | alex_2008 | 선물상자 (IOI15_boxes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "boxes.h"
#include <iostream>
#include <vector>
#include <algorithm>
typedef long long ll;
using namespace std;
ll delivery(int n, int k, int l, vector <int> pos) {
vector <int> a, b;
vector <ll> dp_a, dp_b;
for (int i = 0; i < n; i++) {
if (pos[i] <= (l / 2)) a.push_back(pos[i]);
else b.push_back(pos[i]);
}
reverse(b.begin(), b.end());
for (auto &it : b) {
it = l - it;
}
dp_a.resize((int)a.size());
dp_b.resize((int)b.size());
for (int i = 0; i < (int)dp_a.size(); i++) {
if (i < k) {
dp_a[i] = 2 * a[i];
}
else {
dp_a[i] = dp_a[i - k] + 2 * a[i];
}
}
for (int i = 0; i < (int)dp_b.size(); i++) {
if (i < k) {
dp_b[i] = 2 * b[i];
}
else {
dp_b[i] = dp_b[i - k] + 2 * b[i];
}
}
int ind1 = 0, ind2 = 0;
ll ans = l * 1ll * ((n + k - 1) / k);
ll cur = 0;
for (int i = 1; i <= n; i++) {
if (ind1 == dp_a.size()) {
if (ind2) cur -= dp_b[ind2 - 1];
cur += dp_b[ind2];
ind2++;
}
else if (ind2 == dp_b.size()) {
if (ind1) cur -= dp_a[ind1 - 1];
cur += dp_a[ind1];
ind1++;
}
else {
ll diff1 = dp_a[ind1];
if (ind1) diff1 -= dp_a[ind1 - 1];
ll diff2 = dp_b[ind2];
if (ind2) diff2 -= dp_b[ind2 - 1];
if (diff1 <= diff2) {
cur += diff1;
ind1++;
}
else {
cur += diff2;
ind2++;
}
}
ans = min(ans, ((n - i + k - 1) / k * 1ll * l) + cur);
}
return ans;
}
/*
int main() {
cout << delivery(3, 2, 8, { 1, 2, 5 }) << "\n";
}*/