제출 #841068

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
841068		manhlinh1501	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	42 ms	6716 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;
}

컴파일 시 표준 에러 (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