Submission #1073235

#TimeUsernameProblemLanguageResultExecution timeMemory
1073235IcelastBoxes with souvenirs (IOI15_boxes)C++17
100 / 100
689 ms383556 KiB
#include "boxes.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll maxn = 2*1e5+5, INF = 4e18+9;
ll solve(int n, int k, int B, ll L, vector<int> a){
    vector<ll> b, c;
    b.push_back(0);
    c.push_back(0);
    for(int i = 1; i <= n; i++){
        if(a[i] <= B){
            b.push_back(a[i]);
        }
    }
    for(int i = n; i >= 1; i--){
        if(a[i] > B){
            c.push_back(L-a[i]);
        }
    }
    vector<vector<ll>> sum(2, vector<ll>(k+1, 0)), pf(2, vector<ll>(k+1, 0));
    for(int i = 1; i < (int)b.size(); i++){
        sum[0][i%k] += b[i]*2;
    }
    for(int i = 1; i < (int)c.size(); i++){
        sum[1][i%k] += c[i]*2;
    }
    ll ans = INF;

    int bb = ((int)b.size()-1)%k, cc = ((int)c.size()-1)%k;
    for(int i = 0; i < k; i++){
        int j = (k+cc-(2*k-(bb - i + k))%k)%k;
        ans = min(ans, sum[0][i] +  sum[1][j] + L);
    }
    ans = min(ans, sum[0][bb]+sum[1][cc]);
    if((int)b.size()-1 < k){
        int j = (cc-(k-bb)+k)%k;
        ans = min(ans, L+sum[1][j]);
    }
    if((int)c.size()-1 < k){
        int j = (bb-(k-cc)+k)%k;
        ans = min(ans, L+sum[0][j]);
    }
    return ans;
}
long long delivery(int N, int K, int L, int p[]) {
    int n = N, k = K;
    vector<int> a(n+1);
    for(int i = 1; i <= n; i++){
        a[i] = p[i-1];
    }
    ll ans = INF;
    if(L%2==0){
        ans = min(ans, solve(n, k, L/2, L, a));
        ans = min(ans, solve(n, k, L/2-1, L, a));
    }else{
        ans = min(ans, solve(n, k, L/2, L, a));
    }
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...