Submission #777482

#	Time	Username	Problem	Language	Result	Execution time	Memory
777482		boris_mihov	Boxes with souvenirs (IOI15_boxes)	C++17	10 / 100	1 ms	340 KiB

boxes

#include "boxes.h"
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <vector>

typedef long long llong;
const int MAXN = 10000000 + 10;
const llong INF = 1e18;

int n, k, l;
std::vector <int> left;
std::vector <int> right;
llong prefix[MAXN];
llong suffix[MAXN];

llong delivery(int N, int K, int L, int p[]) 
{
    n = N;
    k = K;
    l = L;

    for (int i = 0 ; i < n ; ++i)
    {
        left.push_back(p[i]);
        right.push_back(l - p[i]);
    }

    for (int i = 0 ; i < n ; ++i)
    {
        prefix[i] = 2 * left[i];
        if (i > 0)
        {
            prefix[i] += prefix[i - 1];
            if (i % k != 0)
            {
                prefix[i] -= 2 * left[i - 1];
            }
        }
    }

    for (int i = n - 1 ; i >= 0 ; --i)
    {
        suffix[i] = 2 * right[i];
        if (i > 0)
        {
            suffix[i] += suffix[i + 1];
            if ((n - 1 - i) % k != 0)
            {
                suffix[i] -= 2 * right[i + 1];
            }
        }
    }

    llong ans = suffix[0];
    for (int i = 0 ; i < n ; ++i)
    {
        ans = std::min(ans, prefix[i] + suffix[i + 1]);
    }
    
    for (int i = 0 ; i + k <= n ; ++i)
    {
        llong curr = l + suffix[i + k];
        if (i > 0) curr += prefix[i - 1];
        ans = std::min(ans, curr);
    }

    return ans;
}

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