# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
420180 | ismoilov | 선물상자 (IOI15_boxes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define IOS ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define all(x) (x).begin(), (x).end()
#define fp(a,i,c) for(int (a) = (i); (a) < (c); (a)++)
#define fpp(a,i,c) for(int (a) = (i); (a) <= (c); (a)++)
#define fm(a,i,c) for(int (a) = (i); (a) > (c); (a)--)
#define fmm(a,i,c) for(int (a) = (i); (a) >= (c); (a)--)
ll delivery(int n, int k, int L, int p[]) {
vector <ll> a, b;
ll ans = 0, l = L;
fp(i,0,n){
if(p[i] <= l-p[i])
a.push_back(p[i]*2);
else
b.push_back((l-p[i])*2);
}
sort(all(a));
sort(all(b));
ll aa = 0, bb = 0;
if(a.size() % k != 0)
aa = a[a.size()-1];
if(b.size() % k != 0)
bb = b[b.size()-1];
if(l >= aa + bb){
int x = k-1;
while(x < a.size())
ans += a[x], x += k;
x = k-1;
while(x < b.size())
ans += b[x], x += k;
if(a.size() % k > 0)
ans += a[a.size()-1];
if(b.size() % k > 0)
ans += b[b.size()-1];
return ans;
}
else{
fp(i,0,k){
if(a.empty()){
if(!b.empty())
b.pop_back();
continue;
}
if(b.empty()){
if(!a.empty())
a.pop_back();
continue;
}
if(a.back() > b.back())
a.pop_back();
else
b.pop_back();
}
int x = k-1;
while(x < a.size())
ans += a[x], x += k;
x = k-1;
while(x < b.size())
ans += b[x], x += k;
if(a.size() % k > 0)
ans += a[a.size()-1];
if(b.size() % k > 0)
ans += b[b.size()-1];
return ans+l;
}
}
int main()
{
IOS;
int n, k, l;
cin >> n >> k >> l;
int p[n];
fp(i,0,n)
cin >> p[i];
cout << delivery(n, k, l, p);
}