Submission #420104

#	Time	Username	Problem	Language	Result	Execution time	Memory
420104		dxz05	Boxes with souvenirs (IOI15_boxes)	C++14	10 / 100	1 ms	204 KiB

boxes

#include "boxes.h"
#include <bits/stdc++.h>

typedef long long ll;

using namespace std;

long long delivery(int N, int K, int L, int p[]) {
    vector<int> v;
    for (int i = 0; i < N; i++){
        if (p[i]) v.push_back(p[i]);
    }
    N = v.size();

    ll ans = 0;
    while (!v.empty()){
        int sz = v.size();
        if (sz <= K){
            int fr = L - v[0], ls = v[sz - 1];

            ans += min({L, fr * 2, ls * 2});

            v.clear();
            break;
        }

        int cost1 = min(L, v[K - 1] * 2);
        int cost2 = min(L, (L - v[sz - K]) * 2);

        if (cost1 <= cost2){
            ans += cost1;
            v.erase(v.begin(), v.begin() + K);
        } else {
            ans += cost2;
            v.erase(v.begin() + sz - K, v.end());
        }

    }

    return ans;
}

Compilation message (stderr)

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

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