Submission #589858

#	Time	Username	Problem	Language	Result	Execution time	Memory
589858		KrisjanisP	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	75 ms	8220 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<ll,ll> ii;
vector<int> find_subset(int l, int u, vector<int> w) {
    ll n = w.size();
    vector<ii> wi;
    for(ll i=0;i<n;i++) wi.push_back({w[i],i});
    sort(wi.begin(),wi.end());
    ll a=0,b=0,s=wi[0].first;
    while(b+1<n&&(s+wi[b+1].first)<=u)
        s+=wi[++b].first;
    //cout<<a<<" "<<b<<" "<<s<<"\n";
    while(b+1<n&&(s+wi[b+1].first-wi[a].first)<=u)
        s+=wi[++b].first-wi[a++].first;
    //cout<<a<<" "<<b<<" "<<s<<"\n";
    if(s>=l&&s<=u)
    {
        vector<int> res;
        for(ll i=a;i<=b;i++)
            res.push_back(wi[i].second);
        return res;
    }
    return vector<int>(0);
}

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