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 <iostream>
#include <bits/stdc++.h>
#define pb push_back
typedef long long ll;
using namespace std;
ll SelectCross (int K, vector<int> I, vector<int> O)
{
ll ats = 0;
int N = I.size();
map<int, vector<int>> lol;
vector<pair<int, int>> visi;
for(int i = 0; i<N; i++)
{
visi.pb({I[i], O[i]});
lol[I[i]].pb(O[i]);
}
sort(visi.begin(), visi.end());
reverse(visi.begin(), visi.end());
int kiek = lol.size();
int kur = 0;
int isviso = 0;
//vector<int> didziausi;
deque<int> didziausi;
for(int i = 0; i<kiek; i++)
{
int dabar = visi[kur].first;
int kiekisI = lol[dabar].size();
isviso += kiekisI;
int k = 0;
int j = 0;
while(j < K && k < kiekisI)
{
if((int)didziausi.size() <= j)
{
didziausi.pb(visi[kur + k].second);
j++; k++;
continue;
}
/*if(visi[kur + k].second <= didziausi[min((int)didziausi.size()-1, K-1)])
{
j = min((int)didziausi.size()-1, K-1) + 1;
continue;
}*/
if(visi[kur + k].second >= didziausi[j])
{
didziausi.insert(didziausi.begin() + j, visi[kur + k].second);
k++;
}
j++;
}
if(isviso < K)
{
kur = isviso;
continue;
}
ll plotas = (ll)dabar*(2*(ll)didziausi[K-1] - (ll)dabar);
ats = max(ats, plotas);
kur = isviso;
}
return ats;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |