Submission #1015613

#	Time	Username	Problem	Language	Result	Execution time	Memory
1015613		Unforgettablepl	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	105 ms	20516 KiB

molecules

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

#define int long long

std::vector<int32_t> find_subset(int32_t l, int32_t u, std::vector<int32_t> w) {
    vector<pair<int,int>> arr;
    for(int i=0;i<w.size();i++)arr.emplace_back(w[i],i);
    sort(arr.begin(),arr.end());
    int n = arr.size();
    int totsum = 0;
    for(auto[a,b]:arr)totsum+=a;
    map<int,int> lookup;
    lookup[0]=n;
    int currsum = totsum;
    for(int i=n-1;i>=0;i--){
        auto it1 = lookup.lower_bound(l-currsum);
        auto it2 = lookup.upper_bound(u-currsum);
        currsum-=arr[i].first;
        lookup[totsum-currsum]=i;
        if(it1==it2)continue;
        int ix = it1->second;
        vector<int32_t> ans;
        for(int j=0;j<=i;j++)ans.emplace_back(arr[j].second);
        for(int j=ix;j<n;j++)ans.emplace_back(arr[j].second);
        return ans;
    }
    auto it1 = lookup.lower_bound(l);
    auto it2 = lookup.upper_bound(u);
    if(it1==it2)return {};
    int ix = it1->second;
    vector<int32_t> ans;
    for(int j=ix;j<n;j++)ans.emplace_back(arr[j].second);
    return ans;
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int32_t, int32_t, std::vector<int>)':
molecules.cpp:8:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    8 |     for(int i=0;i<w.size();i++)arr.emplace_back(w[i],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