Submission #114627

#	Time	Username	Problem	Language	Result	Execution time	Memory
114627		E869120	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	56 ms	6896 KiB

molecules

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

long long N, L, R, A[200009], id[200009];

vector<int> solve(int pos) {
	long long sum = 0;
	for (int i = 0; i < pos; i++) sum += A[i];
	
	if (L <= sum && sum <= R) {
		vector<int> X; for (int i = 0; i < pos; i++) X.push_back(id[i]);
		return X;
	}
	
	for (int i = pos - 1; i >= 0; i--) {
		sum -= A[i];
		sum += A[i + (N - pos)];
		
		if (L <= sum && sum <= R) {
			vector<int> X;
			for (int j = 0; j < i; j++) X.push_back(id[j]);
			for (int j = i + (N - pos); j < N; j++) X.push_back(id[j]);
			return X;
		}
	}
	return vector<int>{};
}

vector<int> find_subset(int l, int u, vector<int> w) {
	vector<pair<int, int>> G; N = w.size(); L = l; R = u;
	for (int i = 0; i < N; i++) G.push_back(make_pair(w[i], i));
	sort(G.begin(), G.end());
	for (int i = 0; i < N; i++) { A[i] = G[i].first; id[i] = G[i].second; }
	
	long long T = 0, C = -1;
	for (int i = N - 1; i >= 0; i--) {
		T += A[i];
		if (T >= l) { C = N - i; break; }
	}
	vector<int> GG = solve(C);
	if (GG.size() >= 1) return GG;
	return vector<int>{};
}

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