Submission #1307728

#	Time	Username	Problem	Language	Result	Execution time	Memory
1307728		ayaz	Detecting Molecules (IOI16_molecules)	C++20	100 / 100	37 ms	5292 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;

#define isz(x) (int)x.size()
#define all(x) (x).begin(),(x).end()

using vi = vector<int>;
using ll = long long;
using vl = vector<ll>;
using pii = pair<int, int>;

vi find_subset(int l, int r, vi w) {
  int n = isz(w);
  vector<array<int, 2>> a(n);
  vl pre(n);
  for (int i = 0; i < n; i++) a[i] = {w[i], i};
  sort(all(a));
  for (int i = 0; i < n; i++) pre[i] = (i == 0 ? a[i][0] : a[i][0] + pre[i - 1]);
  for (int i = 0; i < n; i++) {
    int low = i, high = n - 1;
    auto check = [&](int x) -> bool {
      return (pre[x] - (i == 0 ? 0 : pre[i - 1]) <= r); 
    };
    while (low <= high) {
      int mid = (low + high) >> 1;
      if (check(mid)) {
        low = mid + 1;
      } else {
        high = mid - 1;
      }
    }
    ll sum = pre[low - 1] - (i == 0 ? 0 : pre[i - 1]);
    if (l <= sum && sum <= r) {
      vi idx;
      for (int j = i; j < low; j++) idx.push_back(a[j][1]);
      return idx;
    }
  }
  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