Submission #824107

#	Time	Username	Problem	Language	Result	Execution time	Memory
824107		SoulKnight	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	49 ms	8108 KiB

molecules

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

#define ll long long
#define ln '\n'

const int N = 2e5 + 5;
ll pfx[N], idx[N], n;
vector<ll> vec;
vector<int> ans;

vector<int> find_subset(int l, int u, std::vector<int> w) {
    n = w.size(); vec.resize(n+1);
    iota(idx, idx+n, 0);
    sort(idx, idx+n, [&](int x, int y){return w[x] < w[y];});
    for (int i = 1; i <= n; i++) vec[i] = w[idx[i-1]];

    for (int i = 1; i <= n; i++) pfx[i] = pfx[i-1] + vec[i];

    for (int i = 1; i <= n; i++){
        int j = upper_bound(pfx, pfx+n+1, pfx[i] - l) - pfx - 1;
        if (j == n+1) continue;
        if (l <= pfx[i] - pfx[j] && pfx[i] - pfx[j] <= u){
            for (j++; j <= i; j++) ans.push_back(idx[j-1]);
            break;
        }
    }
    return ans;
}

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