Submission #376060

#	Time	Username	Problem	Language	Result	Execution time	Memory
376060		MilosMilutinovic	Boxes with souvenirs (IOI15_boxes)	C++14	100 / 100	815 ms	215788 KiB

boxes

/**
 *    author:  milos
 *    created: 10.03.2021 17:16:12       
**/
#include "boxes.h"
#include <bits/stdc++.h>
 
using namespace std;
 
long long delivery(int n, int k, int l, int* a) {
  sort(a, a + n);
  vector<long long> pref(n, 0LL);
  for (int i = 0; i < n; i++) {
    if (i >= k) {
      pref[i] = pref[i - k];
    }
    pref[i] += 2 * a[i];
  }
  vector<long long> suff(n, 0LL);
  for (int i = n - 1; i >= 0; i--) {
    if (i + k < n) {
      suff[i] = suff[i + k];
    }
    suff[i] += 2 * (l - a[i]);
  }     
  long long ans = suff[0];
  for (int i = 0; i < n; i++) {
    ans = min(ans, pref[i] + (i < n - 1 ? suff[i + 1] : 0LL)); 
  }
  if (k >= n) {
    ans = min(ans, (long long) l);
  }
  ans = min(ans, suff[k] + l);
  for (int i = 0; i < n; i++) {
    ans = min(ans, pref[i] + (i < n - k - 1 ? suff[i + k + 1] : 0LL) + l);
  }
  return ans;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 15 / 15
• Subtask 4: 15 / 15
• Subtask 5: 20 / 20
• Subtask 6: 30 / 30