Submission #590747

#	Time	Username	Problem	Language	Result	Execution time	Memory
590747		DAleksa	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	53 ms	6316 KiB

molecules

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

using namespace std;

vector<int> find_subset(int l, int r, vector<int> a) {
    vector<int> res;
    int n = a.size();
    vector<int> order(n);
    iota(order.begin(), order.end(), 0);
    sort(order.begin(), order.end(), [&](int i, int j) { return a[i] < a[j]; });
    vector<long long> prefix(n);
    prefix[0] = a[order[0]];
    for(int i = 1; i < n; i++) prefix[i] = prefix[i - 1] + a[order[i]];
    int len = -1;
    for(int i = 0; i < n; i++) {
        if(prefix[i] <= r && prefix[n - 1] - (n - i - 2 >= 0 ? prefix[n - i - 2] : 0) >= l) {
            len = i + 1;
            break;
        }
    }
    if(len == -1) return res;
    if(l <= prefix[len - 1] && prefix[len - 1] <= r) {
        for(int i = 0; i < len; i++) res.push_back(order[i]);
        return res;
    }
    for(int i = 0; i + len < n; i++) {
        if(l <= prefix[i + len] - prefix[i] && prefix[i + len] - prefix[i] <= r) {
            for(int j = i + 1; j <= i + len; j++) res.push_back(order[j]);
            return res;
        }
    }
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:9:24: warning: control reaches end of non-void function [-Wreturn-type]
    9 |     vector<int> order(n);
      |                        ^

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