Submission #479867

#	Time	Username	Problem	Language	Result	Execution time	Memory
479867		ntabc05101	Boxes with souvenirs (IOI15_boxes)	C++14	100 / 100	664 ms	333120 KiB

boxes

#include "boxes.h"
#include<bits/stdc++.h>
using namespace std;
 
#define taskname "CAKES"
 
const int mxN = 1000000;
 
long long delivery(int n, int k, int l, int P[]) {
  vector<int> p;
  for (int i = 0; i < n; i++) {
    auto &x = P[i];
    if (x != 0) {
      p.push_back(x);
    }
  }
  n = p.size();
  
  //return 0;
 
  vector<long long> dp_cw(n), dp_ccw(n);
  for (int i = 0; i < n; i++) {
    if (i < k) {
      dp_cw[i] = min(l, 2 * p[i]);
    }
    else {
      dp_cw[i] = min(l, 2 * p[i]) + dp_cw[i - k];
    }
  }
  for (int i = n - 1; i >= 0; i--) {
    if (i >= n - k) {
      dp_ccw[i] = min(l, 2 * (l - p[i]));
    }
    else {
      dp_ccw[i] = min(l, 2 * (l - p[i])) + dp_ccw[i + k];
    }
  }
  
  long long res = min(dp_cw[n - 1], dp_ccw[0]);
  for (int i = 1; i < n; i++) {
    res = min(res, dp_cw[i - 1] + dp_ccw[i]);
  }
  return res;
}

Compilation message (stderr)

boxes.cpp: In function 'long long int delivery(int, int, int, int*)':
boxes.cpp:17:13: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   17 |   n = p.size();
      |       ~~~~~~^~

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