Submission #1024119

#	Time	Username	Problem	Language	Result	Execution time	Memory
1024119		vanea	Detecting Molecules (IOI16_molecules)	C++14	9 / 100	1 ms	412 KiB

molecules

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

const ll INF = 1e18;

vector<int> find_subset(int l1, int u1, vector<int> w2) {
    vector<pair<ll, int>> w;
    ll l = l1, u = u1;
    for(int i = 0; i < w2.size(); i++) {
        w.push_back({w2[i], i});
    }
    sort(w.begin(), w.end());
    set<array<ll, 2>, greater<array<ll, 2>>> s;
    multiset<pair<ll, int>> w1;
    int n = w.size();
    s.insert({0, 0});
    ll ans = 0;
    for(int i = 0; i < n; i++) {
        if(w[i].first > u) continue;
        w1.insert({w[i].first, w[i].second});
        int k = w[i].first;
        auto it = s.lower_bound({u-k, INF});
        if(it != s.end()) {
            ans = max(ans, k+(*it)[0]);
            s.insert({(*it)[0]+k, k});
        }
        if(it != s.end() && (*it)[0] != 0) s.insert({w[i].first, w[i].first});
    }
    if(ans < l) return {};
    vector<int> res;
    while(ans != 0) {
        auto it = s.lower_bound({ans, INF});
        ll curr = (*it)[1];
        ans -= curr;
        auto it1 = w1.lower_bound({curr, INF});
        res.push_back((*it1).second);
        w1.erase(it1);
    }
    sort(res.begin(), res.end());
    return res;
}
/*
int main()
{
    int l = 16, u = 16;
    vector<int> v = {5, 5, 5, 1};
    auto ans = find_subset(l, u, v);
    for(auto it : ans) {
        cout << it << ' ';
    }
}*/

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:11:22: 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 < w2.size(); i++) {
      |                    ~~^~~~~~~~~~~

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