Submission #744371

#	Time	Username	Problem	Language	Result	Execution time	Memory
744371		Dan4Life	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	53 ms	8240 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
#define all(a) a.begin(),a.end()
using ll = long long;
const int mxN = (int)2e5+10;
array<ll,2> v[mxN];
ll pre[mxN];

vector<int> find_subset(int L, int R, vector<int> w) {
    int n = (int)w.size();
    for(int i = 0; i < n; i++) v[i]={w[i],i};
    sort(v,v+n);
    for(int i = 0; i < n; i++)
        pre[i+1] = pre[i]+v[i][0];
    vector<int> ans;
    for(int i = 0, j = 0; i < n; i++){
        while(j<=n and pre[j]-pre[i] < L) j++;
        if(j==n+1) break;
        ll sum = pre[j]-pre[i];
        if(L<=sum and sum<=R){
            for(int k = i; k < j; k++)
                ans.push_back((int)v[k][1]);
            return ans;
        }
    }
    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