Submission #754869

#	Time	Username	Problem	Language	Result	Execution time	Memory
754869		alexander707070	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	58 ms	7228 KiB

molecules

#include <bits/stdc++.h>
 
using namespace std;
 
int n;
long long pref[200007];
pair<int,int> a[200007];
 
vector<int> answer(int l,int r){
    vector<int> res;
    for(int i=l;i<=r;i++){
        res.push_back(a[i].second);
    }
    return res;
}
 
 
vector<int> find_subset(int l,int u,vector<int> w){
    n=w.size();
    for(int i=1;i<=n;i++){
        a[i]={w[i-1],i-1};
    }

    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++){
        pref[i]=pref[i-1]+a[i].first;
    }

    for(int i=1;i<=n;i++){
        int lt=0, rt=i, mid;
        while(lt+1<rt){
            mid=(lt+rt)/2;
            if(pref[i]-pref[mid]>=l){
                lt=mid;
            }else{
                rt=mid;
            }
        }
        if(pref[i]-pref[lt]>=l and pref[i]-pref[lt]<=u){
            return answer(lt+1,i); 
        }
    }
    
    return {};
}

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