Submission #977964

#	Time	Username	Problem	Language	Result	Execution time	Memory
977964		kilkuwu	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	36 ms	4812 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
  // W_max - W_min <= u - l
  int n = w.size();

  std::vector<int> ord(n);
  std::iota(ord.begin(), ord.end(), 0);
  std::sort(ord.begin(), ord.end(), [&](int i, int j) {
    return w[i] < w[j];
  });

  if (w[ord[0]] > u) return {};
  
  int64_t sum = 0;
  int j = 0;
  for (int i = 0; i < n; i++) {
    sum += w[ord[i]];
    while (sum > u) {
      sum -= w[ord[j]];
      j++;
    }
    if (sum >= l) {
      return std::vector(ord.begin() + j, ord.begin() + i + 1);
    }
  }

  return {};
}

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