Submission #840199

#TimeUsernameProblemLanguageResultExecution timeMemory
840199sleepntsheepDetecting Molecules (IOI16_molecules)C++17
69 / 100
4 ms852 KiB
#include <algorithm>
#include <utility>
#include <set>
#include <deque>

std::vector<int> find_subset(int l, int u, std::vector<int> w)
{
    int n = w.size();
    std::vector<int> rvcord(n);
    {
        std::vector<std::pair<int, int>> incw(n);
        for (int i = 0; i < n; ++i) incw[i] = {w[i], i};
        std::sort(incw.begin(), incw.end());
        for (int i = 0; i < n; ++i) rvcord[i] = incw[i].second;
        std::sort(w.begin(), w.end());
    }
    std::sort(w.begin(), w.end());
    int s = 0, i;
    std::multiset<int> sub;
    std::deque<int> lo, hi;
    for (i = 0; i < (int)w.size(); ++i)
        if (s + w[i] <= u) s += w[i], lo.push_back(i), sub.insert(i);
        else break;
    for (int j = i; j < (int)w.size(); ++j) hi.push_front(j);
    while (s < l && lo.size() && hi.size())
    {
        int d = w[hi.front()] - w[lo.front()];
        if (d <= 0) break;
        s += d;
        sub.erase(sub.find(lo.front())); sub.insert(hi.front());
        lo.push_back(hi.front());
        hi.pop_front(); lo.pop_front(); 
    }
    if (s < l) return std::vector<int>();
    std::vector<int> result(lo.begin(), lo.end());
    for (auto &x : result) x = rvcord[x];
    return result;
}
#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...