Submission #1146716

#	Time	Username	Problem	Language	Result	Execution time	Memory
1146716		KickingKun	Detecting Molecules (IOI16_molecules)	C++20	100 / 100	33 ms	4424 KiB

molecules

#include <bits/stdc++.h>
using namespace std;

#define ll long long

vector <int> find_subset(int l, int u, vector <int> w) {
	vector <int> ind(w.size()); iota(ind.begin(), ind.end(), 0);
	sort (ind.begin(), ind.end(), [&] (int u, int v) {
		return w[u] < w[v];
	});
	
	vector <ll> pref(w.size());
	pref[0] = w[ind[0]];
	for (int i = 1; i < w.size(); i++)
		pref[i] = pref[i - 1] + w[ind[i]];
	
	auto sum = [&] (int l, int r) {
		if (l == 0) return pref[r];
		return pref[r] - pref[l - 1];
	};
	
	for (int k = 1; k <= w.size(); k++) {
		ll miSum = pref[k - 1], maSum = sum(w.size() - k, w.size() - 1);
		if (miSum <= u && maSum >= l) {
			for (int i = k - 1; i < w.size(); i++) {
				ll S = sum(i - k + 1, i);
				// change at most max - min <= u - l
				if (l <= S && S <= u) {
					vector <int> res;
					for (int j = i - k + 1; j <= i; j++)
						res.emplace_back(ind[j]);
					return res;
				}
			}
		}
	}
	return {};
}

Compilation message (stderr)

molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

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