Submission #241187

#	Time	Username	Problem	Language	Result	Execution time	Memory
241187		crossing0ver	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	70 ms	6252 KiB

molecules

#include<bits/stdc++.h>
#include "molecules.h"
using namespace std;
vector<int> answer(deque<int> v) {
	vector<int> ans;
	ans.clear();
	for (int i = 0; i < v.size(); i++) {
		ans.push_back(v[i]);
	}
	return ans;
}
vector<int> find_subset(int l, int u, vector<int> w1) {
	int n = w1.size();
	vector<pair<int,int> > w;
	for (int i = 0; i < w1.size(); i++) w.push_back({w1[i],i});
	sort(w.begin(),w.end());
	long long tot = 0;
	deque<int> v;
	vector<int> ans;
	if (w[0].first > u) return ans;
	for (int i = 0; i < n; i++) {
		if ( tot + w[i].first <= u) {
			tot+=w[i].first;
			v.push_back(w[i].second); 
		}	
	}
	if (tot >= l && tot <= u) return answer(v);
	int in = (v.size() > 0 ?(int)v.size()-1 : -1);
	for (int i = in + 1; i < n; i++) {
		/*if ( tot + w[i].first >= l &&  tot + w[i].first <= u) {
			v.push_back(w[i].second);
			return answer(v);
		}*/
		//if (tot + w[i].first > u) {
			tot -= w1[v.front()];
			v.pop_front();
		//}
		tot += w[i].first;
			v.push_back(w[i].second);
		if (tot >= l && tot <= u) return answer(v);
	}
	return ans;
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> answer(std::deque<int>)':
molecules.cpp:7:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < v.size(); i++) {
                  ~~^~~~~~~~~~
molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:15:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < w1.size(); i++) w.push_back({w1[i],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