Submission #1063651

#	Time	Username	Problem	Language	Result	Execution time	Memory
1063651		HappyCapybara	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	46 ms	6492 KiB

molecules

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

#define ll long long

vector<int> find_subset(int l, int u, vector<int> w){
    int n = w.size();
    vector<int> v = w;
    vector<pair<int,int>> fts(n);
    for (int i=0; i<n; i++) fts[i] = {w[i], i};
    sort(v.begin(), v.end());
    sort(fts.begin(), fts.end());
    ll ls = 0, rs = 0;
    int cur = 0;
    while (!(ls <= u && rs >= l)){
        if (cur == n) return vector<int>(0);
        ls += (ll) v[cur];
        rs += (ll) v[n-cur-1];
        cur++;
    }
    int next = cur;
    while (!(l <= ls && ls <= u)){
        ls += (ll) v[next]-(ll) v[next-cur];
        next++;
    }
    vector<int> res(cur);
    for (int i=0; i<cur; i++) res[i] = fts[next-cur+i].second;
    return res;
}

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