Submission #96271

#	Time	Username	Problem	Language	Result	Execution time	Memory
96271		figter001	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	99 ms	7272 KiB

molecules

#include <bits/stdc++.h>
#include "molecules.h"

using namespace std;

typedef long long ll;
const int N = 2e5+50;

ll sum[N];

vector<int> find_subset(int l, int u, vector<int> w) {
	vector<int> ans;
	vector<pair<int,int>> v;
	for(int i=0;i<w.size();i++)v.push_back({w[i],i});
	sort(v.begin(),v.end());
	sum[0] = v[0].first;
	for(int i=1;i<v.size();i++)sum[i] = sum[i-1] + v[i].first;
	for(int i=0;i<v.size();i++){
		int md,lo=i,hi=v.size()-1,to=-1;
		while(lo <= hi){
			md = (lo+hi)/2;
			ll cur = sum[md];
			if(i)cur -= sum[i-1];
			if(cur >= l && cur <= u){
				to = md;
				break;
			}
			if(cur > u)hi = md-1;
			else if(cur < l)lo = md+1;
		}
		if(to != -1){
			for(int j=i;j<=to;j++)ans.push_back(v[j].second);
			break;
		}
	}
    return ans;
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:14:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<w.size();i++)v.push_back({w[i],i});
              ~^~~~~~~~~
molecules.cpp:17:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=1;i<v.size();i++)sum[i] = sum[i-1] + v[i].first;
              ~^~~~~~~~~
molecules.cpp:18:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<v.size();i++){
              ~^~~~~~~~~

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