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>
#define ll long
using namespace std;
const ll maxn = 2*1e5+5, INF = 4e18+9;
ll solve(int n, int k, int B, int L, vector<int> a){
vector<ll> b(1, 0), c(1, 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 < b.size(); i++){
sum[0][i%k] += b[i]*2;
}
for(int i = 1; i < c.size(); i++){
sum[1][i%k] += c[i]*2;
}
ll ans = INF;
for(int i = 0; i < k; i++){
int j = (1e7*k+c.size()-1-(b.size()-1 - i + 1e7*k)%k)%k;
ans = min(ans, sum[0][i] + sum[1][j] + L);
}
int bb = (b.size()-1)%k, cc = (c.size()-1)%k;
ans = min(ans, sum[0][bb]+sum[1][cc]);
if(b.size()-1 < k){
int j = (1e7*k+c.size()-1-(1e7*k-(b.size()-1)-1+k)%k)%k;
ans = min(ans, L+sum[1][j]);
}
if(c.size()-1 < k){
int j = (1e7*k+b.size()-1-(1e7*k-(c.size()-1)-1+k)%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(n%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;
}
Compilation message (stderr)
boxes.cpp: In function 'long int solve(int, int, int, int, std::vector<int>)':
boxes.cpp:19:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
19 | for(int i = 1; i < b.size(); i++){
| ~~^~~~~~~~~~
boxes.cpp:22:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
22 | for(int i = 1; i < c.size(); i++){
| ~~^~~~~~~~~~
boxes.cpp:27:30: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'double' may change value [-Wconversion]
27 | int j = (1e7*k+c.size()-1-(b.size()-1 - i + 1e7*k)%k)%k;
| ~~~~~~^~
boxes.cpp:27:47: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'double' may change value [-Wconversion]
27 | int j = (1e7*k+c.size()-1-(b.size()-1 - i + 1e7*k)%k)%k;
| ~~~~~~~~~~~^~~
boxes.cpp:27:59: error: invalid operands of types 'double' and 'int' to binary 'operator%'
27 | int j = (1e7*k+c.size()-1-(b.size()-1 - i + 1e7*k)%k)%k;
| ~~~~~~~~~~~~~~~~~~~~~~~~^~
| | |
| double int
boxes.cpp:30:26: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
30 | int bb = (b.size()-1)%k, cc = (c.size()-1)%k;
| ~~~~~~~~~~~~^~
boxes.cpp:30:47: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
30 | int bb = (b.size()-1)%k, cc = (c.size()-1)%k;
| ~~~~~~~~~~~~^~
boxes.cpp:32:19: warning: comparison of integer expressions of different signedness: 'std::vector<long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
32 | if(b.size()-1 < k){
| ~~~~~~~~~~~^~~
boxes.cpp:33:30: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'double' may change value [-Wconversion]
33 | int j = (1e7*k+c.size()-1-(1e7*k-(b.size()-1)-1+k)%k)%k;
| ~~~~~~^~
boxes.cpp:33:51: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'double' may change value [-Wconversion]
33 | int j = (1e7*k+c.size()-1-(1e7*k-(b.size()-1)-1+k)%k)%k;
| ~~~~~~~~~^~~
boxes.cpp:33:59: error: invalid operands of types 'double' and 'int' to binary 'operator%'
33 | int j = (1e7*k+c.size()-1-(1e7*k-(b.size()-1)-1+k)%k)%k;
| ~~~~~~~~~~~~~~~~~~~~~~~~^~
| | |
| | int
| double
boxes.cpp:36:19: warning: comparison of integer expressions of different signedness: 'std::vector<long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
36 | if(c.size()-1 < k){
| ~~~~~~~~~~~^~~
boxes.cpp:37:30: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'double' may change value [-Wconversion]
37 | int j = (1e7*k+b.size()-1-(1e7*k-(c.size()-1)-1+k)%k)%k;
| ~~~~~~^~
boxes.cpp:37:51: warning: conversion from 'std::vector<long int>::size_type' {aka 'long unsigned int'} to 'double' may change value [-Wconversion]
37 | int j = (1e7*k+b.size()-1-(1e7*k-(c.size()-1)-1+k)%k)%k;
| ~~~~~~~~~^~~
boxes.cpp:37:59: error: invalid operands of types 'double' and 'int' to binary 'operator%'
37 | int j = (1e7*k+b.size()-1-(1e7*k-(c.size()-1)-1+k)%k)%k;
| ~~~~~~~~~~~~~~~~~~~~~~~~^~
| | |
| | int
| double