This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "boxes.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
ll delivery(int n, int k, int L, int p[]) {
vector<ll>pref(n,0),suff(n,0);
vector<vector<int>>c(k);
for(int i=0;i<n;i++)c[i%k].push_back(i);
for(int i=0;i<k;i++){
int s = c[i].size();
for(int j=0;j<s;j++){
pref[c[i][j]] = 2*p[c[i][j]];
if(j>=k)pref[c[i][j]]+=pref[c[i][j-k]];
}
for(int j=s-1;j>=0;j--){
suff[c[i][j]] = 2LL*(L-p[c[i][j]]);
if(j+k<s)suff[c[i][j]]+=suff[c[i][j+k]];
}
}
ll ans = 1e18;
ans = min(ans,suff[0]);
ans = min(ans,pref[n-1]);
for(int i=0;i+1<n;i++)ans = min(ans,pref[i] + suff[i+1]);
for(int i=k-1;i<n;i++){
ll cost = L;
if(i-k>=0)cost+=pref[i-k];
if(i+1<n)cost+=suff[i+1];
ans = min(ans,cost);
}
return ans;
}
/*
int main()
{
int n,k,l;
cin>>n>>k>>l;
int p[n];
for(int i=0;i<n;i++)cin>>p[i];
cout<<delivery(n,k,l,p);
}*/
Compilation message (stderr)
boxes.cpp: In function 'll delivery(int, int, int, int*)':
boxes.cpp:11:20: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
11 | int s = c[i].size();
| ~~~~~~~~~^~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |