Submission #319593

#	Time	Username	Problem	Language	Result	Execution time	Memory
319593		knon0501	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	52 ms	6360 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
std::vector<int> find_subset(int l, int u, std::vector<int> w) {
    vector<int> ans;
    vector<pair<int,int>> a;
    for(int i=0 ; i<w.size() ; i++)a.push_back({w[i],i});
    sort(a.begin(),a.end(),[&](pair<int,int> x,pair<int,int> y){
            return x.first>y.first;
         });
    int idx=0;
    for(auto k: w){
        if(k>=l && k<=u){
            ans={idx};
            return ans;
        }
        idx++;
    }
    long long s=0;

    int n=w.size();

    int j=0;

    for(int i=0 ; i<n ; i++){
        while(s<l && j<n){
            s+=a[j].first;
            j++;
        }
        if(s<=u && s>=l){
            for(int k=i ; k<j ; k++)ans.push_back(a[k].second);
            sort(ans.begin(),ans.end());
            return ans;
        }

        s-=a[i].first;
    }
    return {};
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:7:20: 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<w.size() ; i++)a.push_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