Submission #847417

#	Time	Username	Problem	Language	Result	Execution time	Memory
847417		vjudge1	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	56 ms	8896 KiB

molecules

#include <bits/stdc++.h>
using namespace std;
 
using i64 = long long;
using pii = pair<int,int>;
using pli = pair<i64,int>;
 
#define sz(a) (int)a.size()
#define eb emplace_back
#define all(a) a.begin(),a.end()
 
int search(int first,int second,int k,vector<pli> &a){
	int l=first,r=second;
	int pos=-1;
	while(l<=r){
		int m=l+r>>1;
		if(a[m].first-a[first-1].first>=k){
			pos=m;
			r=m-1;
		}
		else l=m+1;
	}
	return pos;
}
 
vector<int> find_subset(int l, int r, vector<int> a){
	int n = sz(a);
	vector<pii> res;
	for(int i=0;i<n;i++)
		res.eb(a[i],i);
	sort(all(res));
	vector<pli> sum(n+1,{0,0});
	for(int i=0;i<n;i++){
		sum[i+1].first=sum[i].first+res[i].first;
		sum[i+1].second=res[i].second;
	}
	vector<int> ans;
	for(int i=1;i<=n;i++){
		int pos=search(i,n,l,sum);
		if(pos==-1)
			continue;
		if(sum[pos].first-sum[i-1].first<=r){
			for(int j=i;j<=pos;j++)
				ans.eb(sum[j].second);
			break;
		}
	}
	sort(all(ans));
	return ans;
}

Compilation message (stderr)

molecules.cpp: In function 'int search(int, int, int, std::vector<std::pair<long long int, int> >&)':
molecules.cpp:16:10: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   16 |   int m=l+r>>1;
      |         ~^~

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