Submission #425398

#TimeUsernameProblemLanguageResultExecution timeMemory
425398dreezyDetecting Molecules (IOI16_molecules)C++17
0 / 100
1 ms332 KiB
#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];
	}
	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;
}


/***************/
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...