Submission #425399

#	Time	Username	Problem	Language	Result	Execution time	Memory
425399		dreezy	Detecting Molecules (IOI16_molecules)	C++17	0 / 100	1 ms	332 KiB

molecules

#include <bits/stdc++.h>
#include "molecules.h"
using namespace std;
#define pb push_back
#define pi pair<int,int>
#define ll long long
#define mp make_pair
vector<int> find_subset(int l, int u, vector<int> ww) {
	
	
	int n = ww.size();
	vector<ll> w(n);
	pi sorted[n];
	for(int i =0; i<n ;i++){
		w[i] = ww[i];
		sorted[i] = {ww[i], i};
	} 
	
	sort(sorted, sorted + n);
	
	int start = lower_bound(sorted, sorted + n, mp(u + 1, -1)) - sorted;
	start--;

	set<pair<int, pair<int,int>>> pos;//val, w, ind
	set<int> nums;
	
	int num_ans  = -1;
	vector<int> ans;
	
	vector<ll> presum(n);
	presum[start] = sorted[start].first;
	for(int s = start-1; s >=0; s--){
		presum[s] = presum[s+1] + sorted[s].first;
	}
	for(; start >=0; start--){
		int i = sorted[start].second;
		//if( cursum + weights[i].w > u) continue;
		
		if(w[i] >=l && w[i] <= u){
			return vector<int>(1, i);
		}
		set<int> ins;
		
		bool won = false;
		for(pair<int,pair<int,int>> val : pos){
			int num = val.first;
			if(num + w[i] >= l && num+w[i] <= u){
				num_ans = num;
				ans.pb(i);
				won = true;
				break;
			}
			
			if(num + w[i] < l ){
				ins.insert(num+w[i]);
			}
		}
	
		if(won ) break;
		
		if(presum[start] + w[i] >= l and nums.find(w[i])== nums.end()){
			pos.insert({w[i], {w[i], i}});
			nums.insert(w[i]);
		}
		
			
		for(int val : ins){
			if(presum[start] + w[i] >= l and nums.find(val)== nums.end()){
				pos.insert({val, {w[i], i}});
				nums.insert(val);
			}
				
		}
	}
	
	if(num_ans == -1)  return  vector<int>();
	
	while(num_ans != 0){
		
		pair<int, pi> val = *lower_bound(pos.begin(), pos.end(), mp(num_ans, mp(-1, -1)));
		ans.pb(val.second.second);
		num_ans = num_ans - val.second.first;
	}
	
	sort(ans.begin(), ans.end());
	return ans;
}


/***************/

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