Submission #240893

#TimeUsernameProblemLanguageResultExecution timeMemory
240893tleontest1Boxes with souvenirs (IOI15_boxes)C++14
10 / 100
7 ms384 KiB
#include "boxes.h"
#pragma GCC optimize ("O3")
#pragma GCC target ("sse4")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target ("avx2")
 
#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long lo;
typedef pair< lo,lo > PII;
 
#define fi first
#define se second
#define mp make_pair
#define endl "\n"
#define pb push_back
#define fio() ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
#define FOR for(int i=1;i<=n;i++)
#define mid ((start+end)/2)
#define ort ((bas+son)/2)
 
const lo inf = 1000000000000000000;
const lo KOK = 100000;
const lo LOG = 30;
const lo li = 500005;
const lo mod = 1000000007;
 
int m,b[li],flag,t;
lo cev;
string s;
multiset<lo> st,st1;
 
long long delivery(int n, int k, int l, int a[]) {
	for(int i=0;i<n;i++){
		cev+=min(l-a[i],a[i])*2;
	}
	if(k==1)return cev;
	cev=0;
	for(int i=0;i<n;i++){
		if(a[i]<l/2)st.insert(a[i]);
		else st1.insert(a[i]);
	}
	//~ cout<<st1.size()<<endl;
	auto itt=st.end();
	int sayy=0;
	while(itt!=st.begin() && sayy<k){sayy++;itt--;}
	if(sayy<k){
		while(1){
			lo say=0;
			auto it=itt;
			if(it==st.begin())flag=1;
			else it--;
			cev+=2**itt;
			//~ cout<<*itt<<endl;
			st.erase(itt);
			if(flag)break;
			itt=it;
			while(say<k-1){say++;auto it=itt;if(it==st.begin())flag=1;else it--;st.erase(itt);if(flag)break;itt=it;}
			if(flag)break;
		}
	}
	//~ cout<<cev<<endl;
	flag=0;
	itt=st1.begin();
	sayy=0;
	flag=0;
	
	while(itt!=st1.end() && sayy<k-1){sayy++;itt++;}
	while(1){
		lo say=0;
			auto it=itt;
			if(it==st1.end())break;
			else it++;
			cev+=2*(l-*itt);
			//~ cout<<*itt<<endl;
			st1.erase(itt);
			if(flag)break;
			itt=it;
			while(say<k-1){say++;auto it=itt;if(it==st1.end())break;else it++;st1.erase(itt);if(flag)break;itt=it;}
			if(it==st1.end())break;
			if(flag)break;
		
	}
	//~ cout<<cev<<endl;
	lo mn=inf;
	auto it=st.end();
	if(it!=st.begin()){
		it--;
		lo at=*it*2;
		it=st1.begin();
		if(it!=st1.end()){
			//~ it--;
			at+=(l-*it)*2;
		}
		
		//~ cout<<at<<endl;
		mn=min(mn,cev+at);
	}
	it=st1.begin();
	if(it!=st1.end()){
		//~ it--;
		int at=(l-*it)*2;
		it=st.end();
		if(it!=st.begin()){
			it--;
			at+=*it*2;
		}
		//~ cout<<at<<endl;
		mn=min(mn,cev+at);
	}
	//~ cout<<mn<<endl;
	//~ cout<<st1.size()<<endl;
	//~ if(st.size() || st1.size()){
		mn=min(mn,cev+(((lo)st.size()+(lo)st1.size()+k-1)/k*l));
		//~ cout<<mn<<endl;
		lo tut=0;
		for(auto it1=st.begin();it1!=st.end();it1++){
			tut++;
			lo tut1=0;
			auto it2=st1.end();
			if(it2==st1.begin())break;
			it2--;
			for(;;it2--){
				tut1++;
				mn=min(mn,cev+*it1*2+(l-*it2)*2+(((lo)st.size()-tut+(lo)st1.size()-tut1+k-1)/k*l));
				//~ cout<<mn<<endl;
				if(it2==st1.begin())break;
			}
		}
	//~ }
    return mn;
}

Compilation message (stderr)

boxes.cpp: In function 'long long int delivery(int, int, int, int*)':
boxes.cpp:60:30: warning: declaration of 'it' shadows a previous local [-Wshadow]
    while(say<k-1){say++;auto it=itt;if(it==st.begin())flag=1;else it--;st.erase(itt);if(flag)break;itt=it;}
                              ^~
boxes.cpp:52:9: note: shadowed declaration is here
    auto it=itt;
         ^~
boxes.cpp:81:30: warning: declaration of 'it' shadows a previous local [-Wshadow]
    while(say<k-1){say++;auto it=itt;if(it==st1.end())break;else it++;st1.erase(itt);if(flag)break;itt=it;}
                              ^~
boxes.cpp:73:9: note: shadowed declaration is here
    auto it=itt;
         ^~
boxes.cpp:104:17: warning: conversion to 'int' from 'long long int' may alter its value [-Wconversion]
   int at=(l-*it)*2;
          ~~~~~~~^~
boxes.cpp:108:12: warning: conversion to 'int' from 'long long int' may alter its value [-Wconversion]
    at+=*it*2;
            ^
#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...