Submission #1307394

#TimeUsernameProblemLanguageResultExecution timeMemory
1307394mikasaDetecting Molecules (IOI16_molecules)C++20
9 / 100
1 ms348 KiB
#include "molecules.h"

#include <map>
#include <algorithm>
#define int32_t int

using namespace std;

std::vector<int32_t> find_subset(int32_t l, int32_t u, std::vector<int32_t> w) {
    vector<int> del;
    long long sum=0;for (int el:w)sum+=el*1ll;
    map<int,vector<int>> mp;
    for (int i=0;i<w.size();++i) {
        mp[w[i]].push_back(i);
    }
    sort(begin(w),end(w));
    int cur=w.size()-1;
    while(sum>u&&cur>=0) {
        while(cur>=0&&sum-w[cur]<l) --cur;
        if (cur<0) {break;}
        del.push_back(mp[w[cur]].back()),mp[w[cur]].pop_back(),sum-=w[cur];
    }
    if (sum>u||sum<l) return {};
    sort(begin(del),end(del));
    cur=0;
    vector<int32_t> res;
    for (int i=0;i<w.size();++i) {
        if (cur<del.size()&&i==del[cur])++cur;
        else res.push_back(i);
    }
    return res;
}

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
      |         ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...