Submission #602638

#	Time	Username	Problem	Language	Result	Execution time	Memory
602638		georgievskiy	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	49 ms	5544 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long ;

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
    int n = w.size();
    vector<pair<int,int>> t(n);
    for (int i = 0; i < n; i++) t[i] = {w[i], i};
    sort(t.begin(), t.end());
    ll s = 0;
    if (u < t[0].first)
        return vector<int>();

    for (int len = 1; len <= n; len++) {
        s += t[len-1].first;
        if (s >= l && s <= u) {
            vector<int> ans;
            for (int i = 0; i < len; i++) ans.push_back(t[i].second);
            return ans;
        }
        if (s > u) {
            s -= t[len-1].first; len--;
            for (int i = len; i < n; i++) {
                s += t[i].first - t[i - len].first;
                if (s >= l && s <= u) {
                    vector<int> ans;
                    for (int j = i - len + 1; j <= i; j++) ans.push_back(t[j].second);
                    return ans;
                }
            }
            break;
        }
    }
    return vector<int>();
}

• Subtask 1: 9 / 9
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 15 / 15
• Subtask 5: 23 / 23
• Subtask 6: 31 / 31