# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
806588 | oscar1f | Cake 3 (JOI19_cake3) | C++17 | 0 ms | 0 KiB |
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<bits/stdc++.h>
using namespace std;
#define int long long
using pii=pair<int,int>;
string to_string(string s) { return s; }
template <typename T> string to_string(T v) {
bool first = true;
string res = "[";
for (const auto &x : v) {
if (!first)
res += ", ";
first = false;
res += to_string(x);
}
res += "]";
return res;
}
template <typename A, typename B>
string to_string(pair<A, B> p) {
return "(" + to_string(p.first) + ", " + to_string(p.second) + ")";
}
void dbg_out() { cout << endl; }
template <typename Head, typename... Tail> void dbg_out(Head H, Tail... T) {
cout << ' ' << to_string(H);
dbg_out(T...);
}
#ifdef DEBUG
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif
struct noeud{
int minVal,maxVal,pos;
noeud *filsGauche=nullptr,*filsDroit=nullptr;
vector<int> cumuGauche,cumuVal,cumuDroite;
noeud (vector<int> listeVal,int minVal_,int maxVal_) {
minVal=minVal_;
maxVal=maxVal_;
//dbg(minVal,maxVal,listeVal);
cumuVal.push_back(0);
for (auto x:listeVal) {
cumuVal.push_back(x+cumuVal.back());
}
int taille=listeVal.size();
if (minVal!=maxVal) {
int mid=(minVal+maxVal)/2;
vector<int> listeGauche,listeDroite;
cumuGauche.push_back(0);
cumuDroite.push_back(0);
for (int x:listeVal) {
int estGauche;
if (x<=mid) {
estGauche=1;
listeGauche.push_back(x);
}
else {
estGauche=0;
listeDroite.push_back(x);
}
cumuGauche.push_back(cumuGauche.back()+estGauche);
cumuDroite.push_back(cumuDroite.back()+1-estGauche);
}
if (!listeGauche.empty()) {
filsGauche=new noeud(listeGauche,minVal,mid);
}
if (!listeDroite.empty()) {
filsDroit=new noeud(listeDroite,mid+1,maxVal);
}
}
else {
cumuGauche.resize(taille+1,0);
cumuDroite.resize(taille+1,0);
}
}
pair<int,int> somMax(int deb,int fin,int nbVal) {
if (deb>fin) {
return {0,0};
}
//int nbGauche=cumuGauche[fin+1]-cumuGauche[deb];
//int nbDroite=cumuDroite[fin+1]-cumuDroite[deb];
int debutGauche=cumuGauche[deb];
int finGauche=cumuGauche[fin+1]-1;
int debutDroite=cumuDroite[deb];
int finDroite=cumuDroite[fin+1]-1;
int somDroit=0,cntDroit=0,somGauche=0,cntGauche=0;
if (filsDroit!=nullptr) {
if (finDroite-debDroite+1>nbVal) {
tie(somDroit,cntDroit)=filsDroit->somMax(debutDroite,finDroite,nbVal);
nbVal-=cntDroit;
}
else {
cntDroit=finDroite-debDroite+1;
somDroit=
}
}
if (nbVal>0 and filsGauche!=nullptr) {
tie(somGauche,cntGauche)=filsGauche->somMax(debutGauche,finGauche,nbVal);
}
dbg(somDroit,somGauche,cntDroit,cntGauche);
return {somDroit+somGauche,cntDroit+cntGauche};
}
};
const int INFINI=(int)1000*1000*1000*1000*1000*1000;
int nbPossi,tailleCol,prix,benef,rep;
noeud *racine;
vector<pair<int,int>> possi;
int somMax(int deb,int fin,int nbVal) {
return racine->somMax(deb,fin,nbVal).first;
}
int calcMeil(int deb,int fin) {
return possi[deb].second+possi[fin].second-2*(possi[fin].first-possi[deb].first)+somMax(deb+1,fin-1,tailleCol-2);
}
void calc(int minDeb,int maxDeb,int minFin,int maxFin) {
if (maxDeb<minDeb or maxFin<minFin) {
return;
}
int deb=(minDeb+maxDeb)/2;
pii record={-INFINI,maxFin};
dbg(deb);
for (int i=minFin;i<=maxFin;i++) {
if (i-deb+1>=tailleCol) {
dbg(deb,i);
record=max(record,{calcMeil(deb,i),i});
}
}
rep=max(rep,record.first);
calc(minDeb,deb-1,minFin,record.second);
calc(deb+1,maxDeb,record.second,maxFin);
}
signed main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin>>nbPossi>>tailleCol;
for (int i=0;i<nbPossi;i++) {
cin>>benef>>prix;
possi.push_back({prix,benef});
}
sort(possi.begin(),possi.end());
vector<int> listeCoul;
for (auto [_,coul]:possi) {
listeCoul.push_back(coul);
}
rep=-INFINI;
racine=new noeud(listeCoul,1,1000*1000*1000);
calc(0,nbPossi-1,0,nbPossi-1);
cout<<rep<<endl;
return 0;
}