Submission #801801

#	Time	Username	Problem	Language	Result	Execution time	Memory
801801		_martynas	Boxes with souvenirs (IOI15_boxes)	C++11	100 / 100	588 ms	202776 KiB

boxes

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

using namespace std;

vector<long long> get_cost_vector(int N, int K, int L, int p[]) {
    vector<long long> cost(N);
    for(int i = 0; i < N; i++) {
        cost[i] += p[i];
        if(i-K >= 0) cost[i] += cost[i-K]+p[i-K];
    }
    return cost;
}

long long delivery(int N, int K, int L, int p[]) {
    int zeroes = count(p, p+N, 0);
    N -= zeroes;
    for(int i = 0; i < N; i++) p[i] = p[i+zeroes];
    vector<long long> cost_l = get_cost_vector(N, K, L, p);
    reverse(p, p+N);
    for(int i = 0; i < N; i++) {
        p[i] = (p[i] != 0 ? L-p[i] : p[i]);
    }
    vector<long long> cost_r = get_cost_vector(N, K, L, p);
    long long ans = min(cost_l[N-1]+p[0], cost_r[N-1]+p[N-1]);
    ans = min(ans, min(cost_l[N-1]+(L-p[0]), cost_r[N-1]+(L-p[N-1])));
    reverse(cost_r.begin(), cost_r.end());
    reverse(p, p+N);
    for(int i = 0; i < N; i++) {
        p[i] = (p[i] != 0 ? L-p[i] : p[i]);
    }
    for(int i = 1; i < N; i++) {
        ans = min(ans, cost_l[i-1]+min(p[i-1], L-p[i-1])+cost_r[i]+min(p[i], L-p[i]));
    }
    return ans;
}

Compilation message (stderr)

boxes.cpp: In function 'std::vector<long long int> get_cost_vector(int, int, int, int*)':
boxes.cpp:6:53: warning: unused parameter 'L' [-Wunused-parameter]
    6 | vector<long long> get_cost_vector(int N, int K, int L, int p[]) {
      |                                                 ~~~~^
boxes.cpp: In function 'long long int delivery(int, int, int, int*)':
boxes.cpp:16:23: warning: conversion from 'std::iterator_traits<int*>::difference_type' {aka 'long int'} to 'int' may change value [-Wconversion]
   16 |     int zeroes = count(p, p+N, 0);
      |                  ~~~~~^~~~~~~~~~~

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