Submission #157439

#	Time	Username	Problem	Language	Result	Execution time	Memory
157439		dennisstar	Detecting Molecules (IOI16_molecules)	C++11	100 / 100	68 ms	6252 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int N;
vector<pair<int,int> > pi;
vector<int> ans;
vector<int> find_subset(int l, int u, vector<int> w) {
    int i; N=w.size();
	for (i=0; i<N; i++) pi.push_back({w[i], i});
	sort(pi.begin(), pi.end());
	ll s=0;
	for (i=N-1; i>=0; i--) {
		s+=(ll)pi[i].first;
		ans.push_back(i);
		if (s>=(ll)l) break;
	}
	if (i<0) return vector<int>(0);
	for (; i>=0; i--) {
		if (s<=(ll)u) break;
		s-=(ll)pi[ans[(N-1-i)%ans.size()]].first; s+=(ll)pi[i].first; ans[(N-1-i)%ans.size()]=i;
	}
	if (s<=(ll)u) {
		for (i=0; i<ans.size(); i++) ans[i]=pi[ans[i]].second;
		return ans;
	}
	return vector<int>(0);
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:24:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (i=0; i<ans.size(); i++) ans[i]=pi[ans[i]].second;
             ~^~~~~~~~~~~

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