Submission #687121

#	Time	Username	Problem	Language	Result	Execution time	Memory
687121		viwlesxq	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	97 ms	13572 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
 
using namespace std;
 
typedef int64_t ll;
typedef string str;
 
#define pb push_back
#define pf push_front
#define ppb pop_back
#define ppf pop_front
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define sz(x) (int)x.size()
 
vector <int> find_subset(int l, int r, vector <int> w) {
	int n = sz(w);
	set <int> st;
	ll cur = 0;
	pair <int, int> v[n];
	for (int i = 0; i < n; i++) {
		v[i] = {w[i], i};
	}
	sort(v, v + n);
	int tl = 0, tr = 0;
	while (tr < n) {
		if (cur < l) {
			cur += v[tr].F;
			st.insert(v[tr].S);
			tr++;
		}
		if (cur > r) {
			cur -= v[tl].F;
			st.erase(v[tl].S);
			tl++;
		}
		if (cur >= l && cur <= r) {
			break;
		}
	}
	if (cur >= l && cur <= r) {
		return vector <int> (all(st));
	}
	return {};
}

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