Submission #810823

#	Time	Username	Problem	Language	Result	Execution time	Memory
810823		wortelworm	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	44 ms	5556 KiB

molecules

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

typedef long long ll;
std::vector<int> find_subset(int lower, int upper, std::vector<int> wOriginal) {
    ll sum = 0;
    // size, index
    vector<pair<int, int>> weights(wOriginal.size());
    for (int i = 0; i < wOriginal.size(); i++) {
        weights[i] = {wOriginal[i], i};
    }

    sort(weights.rbegin(), weights.rend());

    // vector<int> past;
    queue<int> past;
    for (int i = 0; i < weights.size(); i++) {
        auto [w, index] = weights[i];
        sum += w;
        past.push(index);

        while (sum > upper) {
            sum -= wOriginal[past.front()];
            past.pop();
        }
        
        if (sum >= lower) {

            vector<int> answer;
            answer.reserve(past.size());
            while (past.size() > 0)
            {
                answer.push_back(past.front());
                past.pop();
            }
            
            return answer;
        }
    }

    return {};
}

Compilation message (stderr)

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