Submission #841067

#	Time	Username	Problem	Language	Result	Execution time	Memory
841067		manhlinh1501	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	45 ms	8500 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++){
		// for(int j=i;j<=n;j++){
			// i64 res=sum[j].first-sum[i-1].first;
			// if(l<=res and res<=r){
				// for(int k=i;k<=j;k++)
					// ans.eb(sum[k].second);
				// sort(all(ans));
				// return ans;
			// }
		// }
		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++)
				// cout<<sum[j].second<<" ";
			// cout<<"\n";
			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:14:10: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   14 |   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