| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 127949 | Sorting | 선물상자 (IOI15_boxes) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
const int N = 1e7 + 7;
vector<long long> v1, v2;
vector<long long> v3, v4;
void output(vector<long long> v){
for(long long &x: v){
cout << x << " ";
}
cout << endl;
}
long long delivery(int n, int k, int l, int *positions){
sort(positions, positions + n);
int idx = 0;
for(; idx < n && positions[idx] <= l / 2; idx++){
v1.push_back(positions[idx]);
}
for(; idx < n; idx++){
v2.push_back(l - positions[idx]);
}
if(!v2.empty()){
reverse(v2.begin(), v2.end());
}
//output(v1);
//output(v2);
long long ans = 0, sum1 = 0, sum2 = 0;
for(int i = -1; true;){
if(i + k >= (int)v1.size()){
for(++i; i < (int)v1.size(); ++i){
v3.push_back(v1[i]);
}
break;
}
i += k;
sum1 += (long long)v1[i] * 2ll;
}
for(int i = -1; true;){
if(i + k >= (int)v2.size()){
for(++i; i < (int)v2.size(); ++i){
v4.push_back(v2[i]);
}
break;
}
i += k;
sum2 += (long long)v2[i] * 2ll;
}
//output(v3);
//output(v4);
ans = sum1 + sum2;
if(v3.empty() || v4.empty()){
if(!v3.empty()){
ans += v3.back() * 2ll;
}
if(!v4.empty()){
ans += v4.back() * 2ll;
}
return ans;
}
if((int)v3.size() + (int)v4.size() <= k){
ans += min(l, v3.back() * 2ll + v4.back() * 2ll);
return ans;
}
int j = (int)v3.size() + (int)v4.size() - k;
ans += v3.back() * 2ll + v4.back() * 2ll;
if((int)v3.size() >= j){
long long ans2 = sum2;
for(int i = (int)v1.size() + j - (int)v3.size() - 1; i >= 0; i -= k){
ans2 += 2ll * v1[i];
}
ans = min(ans, ans2);
}
if((int)v4.size() >= j){
long long ans2 = sum2;
for(int i = (int)v2.size() + j - (int)v4.size() - 1; i >= 0; i -= k){
ans2 += 2ll * v2[i];
}
ans = min(ans, ans2);
}
return ans;
}
/*int positions[N], n, k, l;
int main(){
cin >> n >> k >> l;
for(int i = 0; i < n; i++){
cin >> positions[i];
}
cout << delivery(n, k, l, positions) << "\n";
return 0;
}*/
/*
3 2 8
1 2 5
*/
