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;
//int didziausias = 0;
for(int i = 0; i<N; i++)
{
visi.pb({I[i], O[i]});
lol[I[i]].pb(O[i]);
//didziausias = max(didziausias, I[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;
if(isviso<K)
{
for(int j = 0; j<kiekisI; j++)
{
didziausi.pb(visi[j].second);
}
/*for(auto c : lol[dabar])
{
didziausi.pb(c);
}*/
kur = isviso;
continue;
}
if(didziausi.size() == 0)
{
for(int j = 0; j<min(K, kiekisI); j++)
{
didziausi.pb(visi[kur + j].second);
}
} else
{
/*for(int j = 0; j<min(K, kiekisI); j++)
{
if(didziausi.size()>0)
{
if(visi[kur + j].second > didziausi[K-1])
{
didziausi.pb(visi[kur + j].second);
}
} else
{
didziausi.pb(visi[kur + j].second);
}
}*/
int k = 0;
int j = 0;
while(j < K && k < kiekisI)
{
if(visi[kur + k].second > didziausi[j])
{
didziausi.insert(didziausi.begin() + j, visi[kur + k].second);
k++;
}
j++;
}
}
/*sort(didziausi.begin(), didziausi.end());
reverse(didziausi.begin(), didziausi.end());
if(didziausi.size()>K)
didziausi.erase(didziausi.begin() + K, didziausi.end());*/
ll plotas = (ll)dabar*(2*(ll)didziausi[min((int)didziausi.size()-1, 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... |