# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
388995 | apostoldaniel854 | Detecting Molecules (IOI16_molecules) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define dbg(x) cerr << #x << " " << x << "\n"
#define HOME
#ifndef HOME
#include "molecules.h"
#endif // HOME
std::vector<int> find_subset(int l, int u, std::vector<int> w) {
int n = w.size ();
vector <pair <int, int>> v (n);
for (int i = 0; i < n; i++)
v[i] = {w[i], i};
int lb = 0, rb = 0;
ll sum = v[0].first;
while (rb < n && lb <= rb) {
if (sum >= l && sum <= u) {
vector <int> sol;
for (int i = lb; i <= rb; i++)
sol.push_back (v[i].second);
return sol;
}
if (sum < l)
sum += v[++rb].first;
else
sum -= v[lb++].first;
}
return {};
}
#ifdef HOME
int main() {
int n, l, u;
assert(3 == scanf("%d %d %d", &n, &l, &u));
std::vector<int> w(n);
for (int i = 0; i < n; i++)
assert(1 == scanf("%d", &w[i]));
std::vector<int> result = find_subset(l, u, w);
printf("%d\n", (int)result.size());
for (int i = 0; i < (int)result.size(); i++)
printf("%d%c", result[i], " \n"[i == (int)result.size() - 1]);
}
#endif