Submission #114627

#TimeUsernameProblemLanguageResultExecution timeMemory
114627E869120Detecting Molecules (IOI16_molecules)C++14
100 / 100
56 ms6896 KiB
#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>{};
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...