Submission #1148860

#	Time	Username	Problem	Language	Result	Execution time	Memory
1148860		kvintsekstakord	Detecting Molecules (IOI16_molecules)	C++20	69 / 100	33 ms	4276 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<pair<int, int>> a;
    
    for(int i = 0; i < w.size(); i++){
        a.push_back({w[i], i});
    }
    sort(a.begin(), a.end());
    
    vector<int> pref(w.size()+1);
    pref[0]=0;

    for(int i = 0; i < w.size(); i++) pref[i+1]=pref[i]+a[i].first;

    for(int k = 1; k <= w.size(); k++){    
            if(pref[k]>u || pref[w.size()]-pref[w.size()-k]<l) continue;

            vector<int> ans;
            int start = -1;
            for(int i = k; i <= w.size(); i++){
                if(pref[i]-pref[i-k]<=u &&  pref[i]-pref[i-k]>=l ){
                    start = i;
                }
            }
            if(start==-1) continue;

            for(int i = start; i > start-k; i--){
                ans.push_back(a[i-1].second);
            }
            return ans;
            
    }
    

    return std::vector<int>(0);
}

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: 0 / 31