Submission #107195

#	Time	Username	Problem	Language	Result	Execution time	Memory
107195		popovicirobert	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	187 ms	15196 KiB

molecules

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

using namespace std;

vector<int> find_subset(int l, int u, vector<int> w) {

    vector < pair <int, int> > arr(w.size());
    for(int i = 0; i < w.size(); i++) {
        arr[i] = {w[i], i};
    }

    sort(arr.begin(), arr.end());

    multiset <ll> mst;
    mst.insert(0);

    ll sum = 0;
    ll sl, sr = -1;
    for(auto it : arr) {
        sum += it.first;

        auto cur = mst.lower_bound(sum - u);
        if(cur != mst.end() && *cur <= sum - l) {
            sl = *cur;
            sr = sum;
            break;
        }

        mst.insert(sum);
    }

    vector <int> sol;
    if(sr == -1) {
        return sol;
    }

    sum = 0;

    for(int i = 0; i < arr.size(); i++) {
        sum += arr[i].first;

        if(sum > sl) {
            sol.push_back(arr[i].second);
        }

        if(sum == sr) {
            break;
        }

    }

    return sol;

}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:10:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0; i < w.size(); i++) {
                    ~~^~~~~~~~~~
molecules.cpp:41:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0; i < arr.size(); i++) {
                    ~~^~~~~~~~~~~~

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