Submission #380682

#	Time	Username	Problem	Language	Result	Execution time	Memory
380682		madlogic	Detecting Molecules (IOI16_molecules)	C++17	9 / 100	1 ms	384 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) {
  int n = (int) w.size();
  for (int i = 0; i < n; i++) {
    if (w[i] >= l && w[i] <= u) {
      return {i};
    }
  }
  vector<pair<int, int>> nums;
  for (int i = 0; i < n; i++) {
    if (w[i] <= u) {
      nums.emplace_back(w[i], i);
    }
  }
  sort(nums.begin(), nums.end());
  int cur = 0;
  multiset<pair<int, int>> ms;
  for (auto& [x, y] : nums) {
    cur += x;
    ms.insert({x, y});
    if (cur > u) {
      int remove = u - cur;
      auto lb = ms.lower_bound({remove, -1});
      cur -= (*lb).first;
      ms.erase(lb);
    }
  }
  if (cur < l) {
    return {};
  }
  vector<int> res;
  for (auto s : ms) {
    res.push_back(s.second);
  }
  return res;
}

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