Submission #668375

#	Time	Username	Problem	Language	Result	Execution time	Memory
668375		haxorman	Detecting Molecules (IOI16_molecules)	C++14	0 / 100	0 ms	212 KiB

molecules

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

vector<int> find_subset(int l, int r, vector<int> arr) {
    vector<pair<int,int>> vec;
    for (int i = 0; i < arr.size(); ++i) {
        vec.push_back({arr[i], i + 1});
    }
    sort(vec.begin(), vec.end());
    
    long long sum = 0;
    set<pair<int,int>> check;
    for (int i = 0; i < vec.size(); ++i) {
        while (sum > r) {
            sum -= (*check.rbegin()).first;
            check.erase(*check.rbegin());
        }

        if (sum >= l) {
            vector<int> ans;
            for (auto p : check) {
                ans.push_back(p.second);
            }
            return ans;
        }

        check.insert(vec[i]);
        sum += vec[i].first;
    }
    while (sum > r) {
        sum -= (*check.rbegin()).first;
        check.erase(*check.rbegin());
    }

    if (sum >= l) {
        vector<int> ans;
        for (auto p : check) {
            ans.push_back(p.second);
        }
        return ans;
    }
    return {};
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:7:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    7 |     for (int i = 0; i < arr.size(); ++i) {
      |                     ~~^~~~~~~~~~~~
molecules.cpp:14:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   14 |     for (int i = 0; i < vec.size(); ++i) {
      |                     ~~^~~~~~~~~~~~

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