Submission #239429

#	Time	Username	Problem	Language	Result	Execution time	Memory
239429		lyc	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	71 ms	2936 KiB

molecules

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

#define TRACE(x) cout << #x << " :: " << x << endl;
#define _ << " " <<
#define FOR(i,a,b) for(int i=(a);i<=(b);++i)
#define RFOR(i,a,b) for(int i=(a);i>=(b);--i)
#define SZ(x) ((int)(x).size())
#define ALL(x) (x).begin(),(x).end()

vector<int> find_subset(int l, int u, vector<int> w) {
    int N = SZ(w);
    int ind[N];
    FOR(i,0,N-1){ ind[i] = i; }
    sort(ind,ind+N,[&](int a, int b){ return (w[a] != w[b] ? w[a] < w[b] : a < b); });
    long long sum = 0;
    int x = 0, y;
    for (y = 0; y < N; ++y) {
        sum += w[ind[y]];
        while (sum > u) sum -= w[ind[x++]];
        if (sum >= l) break;
    }
    if (sum < l) return vector<int>();
    else {
        vector<int> ans(y-x+1);
        FOR(i,x,y) ans[i-x] = ind[i];
        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