Submission #772869

#	Time	Username	Problem	Language	Result	Execution time	Memory
772869		Tkm_algo	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	38 ms	6180 KiB

molecules

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

using namespace std;
using ll = long long;
const int N = 5e5 + 10;

vector<int> find_subset(int l, int u, vector<int> w) {
	vector<pair<int, int>> v;
	int n = w.size();
	for (int i = 0; i < n; i++) {
		v.push_back({w[i], i});
	}
	sort(v.begin(), v.end());
	int lf = -1, r = -1;
	ll sm = 0;
	vector<int> ans;
	while (1) {
		if (sm < l && r < n - 1) {
			r++;
			sm += v[r].first;
		} else {
			if (sm > u && lf < n - 1) {
				lf++;
				sm -= v[lf].first;
			} else {
				if (sm >= l && sm <= u) {
					for (int i = lf + 1; i <= r; i++) {
						ans.push_back(v[i].second);
					}
				}
				break;
			}
		}
	}
	return ans;
}

// int main() {
// 	ios::sync_with_stdio(false);
// 	cin.tie(0);
	
// 	int l, u;
// 	cin >> l >> u;
// 	int n;
// 	cin >> n;
// 	vector<int> w;
// 	for (int i = 0; i < n; i++) {
// 		int x;
// 		cin >> x;
// 		w.push_back(x);
// 	}
// 	vector<int> ans = find_subset(l, u, w);
// 	for (auto x : ans) {
// 		cout << x << ' ';
// 	}
// }

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