Submission #135759

#	Time	Username	Problem	Language	Result	Execution time	Memory
135759		Runtime_error_	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	187 ms	21352 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll inf = 2e5+9;
ll n,pre[inf];
pair<ll,ll> a[inf];

vector<int> find_subset(int l, int r, vector<int> w) {
    vector<int> ans;
    n = w.size();
    for(ll i=1;i<=n;i++)
        a[i] = make_pair(w[i-1],i-1);

    sort(a+1,a+1+n);
    for(ll i=1;i<=n;i++)
        pre[i] = pre[i-1]+a[i].first;

    set<pair<ll,ll> > s;

    for(ll i=1;i<=n;i++){

        s.insert(make_pair(pre[i-1],i));
        set< pair<ll,ll> > ::iterator it = s.lower_bound( make_pair( pre[i]-r , 0 ) );

        if( it == s.end() || it->first > pre[i] - l )
            continue;

        for(ll j = it->second ; j <= i ;j++)
            ans.push_back(a[j].second);

        return ans;

        /*for(ll j=i;j<=n;j++){
            if(pre[j] - pre[i-1] >=l && pre[j] - pre[i-1] <= r){
                for(ll z=i;z<=j;z++)
                    ans.push_back(a[z].second-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