Submission #420145

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

boxes

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

typedef long long ll;

using namespace std;

const int MAXN = 1111111;

int dp[MAXN];
ll suff[MAXN];

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

    ll cost = 0, prizes = K;
    for (int i = rg.size() - 1; i >= 0; i--){
        if (prizes){
            prizes--;
            cost += (i < rg.size() - 1 ? rg[i + 1] : L) - rg[i];
            dp[i] = cost;
        } else {
            prizes = K - 1;
            cost += L - rg[i + 1] + L - rg[i];
            dp[i] = cost;
        }
        dp[i] += L - rg[i];
        suff[i] = suff[i + 1] + dp[i];
    }


    cost = 0, prizes = K;
    for (int i = 0; i < lf.size(); i++){
        if (prizes){
            prizes--;
            cost += lf[i] - (i > 0 ? lf[i - 1] : 0);
        } else {
            prizes = K - 1;
            cost += lf[i] - lf[i - 1];
        }
    }

    ll ans = cost * 2 + suff[0];

    if (prizes > 0 && !lf.empty()){
        if (prizes >= rg.size()){
            ans = min(ans, cost + L - lf.back());
        } else {
            ans = min(ans, (cost + rg[prizes - 1] - lf.back()) * 2 + suff[prizes]);
        }
    }

    return ans;
}

Compilation message (stderr)

boxes.cpp: In function 'long long int delivery(int, int, int, int*)':
boxes.cpp:21:15: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   21 |     N = v.size();
      |         ~~~~~~^~
boxes.cpp:24:28: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   24 |     for (int i = rg.size() - 1; i >= 0; i--){
      |                  ~~~~~~~~~~^~~
boxes.cpp:27:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |             cost += (i < rg.size() - 1 ? rg[i + 1] : L) - rg[i];
      |                      ~~^~~~~~~~~~~~~~~
boxes.cpp:28:21: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
   28 |             dp[i] = cost;
      |                     ^~~~
boxes.cpp:32:21: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
   32 |             dp[i] = cost;
      |                     ^~~~
boxes.cpp:40:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |     for (int i = 0; i < lf.size(); i++){
      |                     ~~^~~~~~~~~~~
boxes.cpp:53:20: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |         if (prizes >= rg.size()){
      |             ~~~~~~~^~~~~~~~~~~~

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