Submission #1163111

#	Time	Username	Problem	Language	Result	Execution time	Memory
1163111		boclobanchat	Detecting Molecules (IOI16_molecules)	C++20	100 / 100	50 ms	6584 KiB

molecules

#include"molecules.h"
#include<bits/stdc++.h>
using namespace std;
vector<int> find_subset(int l,int u,vector<int> w)
{
	if(w.size()==1)
	{
		vector<int> ans;
		if(w[0]>=l&&w[0]<=u) ans.push_back(0);
		return ans;
	}
	int idx=0;
	vector< pair<int,int> > x;
	for(auto v:w) x.push_back({v,idx++});
	sort(w.begin(),w.end());
	sort(x.begin(),x.end());
	vector<int> ans;
	vector<long long> pref;
	pref.push_back(0);
	int n=w.size();
	long long sum=0;
	for(int i=0;i<n;i++)
	{
		pref.push_back(sum+=w[i]);
		int pos=upper_bound(pref.begin(),pref.end(),sum-l)-pref.begin()-1;
		if(pos<0) continue;
		if(sum-pref[pos]<=u)
		{
			for(int j=pos+1;j<=i+1;j++) ans.push_back(x[j-1].second);
			sort(ans.begin(),ans.end());
			return ans;
		}
	}
	return ans;
}

Compilation message (stderr)

molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

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