Submission #1204622

#	Time	Username	Problem	Language	Result	Execution time	Memory
1204622		loiiii12358	Detecting Molecules (IOI16_molecules)	C++20	100 / 100	36 ms	6172 KiB

molecules

#include "molecules.h"
#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>> W;
    vector<int32_t> ans,ans2;
    for(int i=0;i<w.size();i++){
        W.push_back({w[i],i});
    }
    sort(W.begin(),W.end());
    int mi=0,ma=0,pl=0,pr=W.size()-1;
    for(int i=0;i<W.size();i++){
        mi+=W[i].first;
        ma+=W[W.size()-1-i].first;
        if(ma>=l&&mi<=u){
            pl=i;
            while(pl>=0&&mi+W[pr].first-W[pl].first<=u){
                ans.push_back(W[pr].second);
                mi+=W[pr--].first-W[pl--].first;
            }
            for(int j=0;j<pl;j++){
                ans.push_back(W[j].second);
            }
            while(mi<l){
                mi+=W[pl+1].first-W[pl].first;
                pl++;
            }
            if(pl>=0){
                ans.push_back(W[pl].second);
            }
            return ans;
        }
    }
    return {};
}

Compilation message (stderr)

molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

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