Submission #380365

#	Time	Username	Problem	Language	Result	Execution time	Memory
380365		ritul_kr_singh	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	64 ms	6256 KiB

molecules

#include <bits/stdc++.h>
#include "molecules.h"
using namespace std;
 
vector<int> find_subset(int lx, int ux, vector<int> w){
  	using ll = long long;
  	ll l = lx, u = ux;
	vector<pair<ll, ll>> a;
	int curr = 0;
	ll _sum = 0, n = w.size();
	for(int i : w) a.emplace_back(i, curr++), _sum += (ll)i;
	sort(a.begin(), a.end());
	ll sum = _sum;
	int j = 0;
	for(int i=-1; i<n; ++i){
		if(i>=0) sum += a[i].first;
		while(j<=i) sum -= a[j++].first;
		while(j<n and sum>u) sum -= a[j++].first;
		if(l<=sum and sum<=u){
			vector<int> ans;
			for(int k=0; k<=i; ++k) ans.push_back(a[k].second);
			for(int k=n-1; k>=j; --k) ans.push_back(a[k].second);
			sort(ans.begin(), ans.end());
			return ans;
		}
	}
	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