Submission #1266604

#	Time	Username	Problem	Language	Result	Execution time	Memory
1266604		thewizardman	Detecting Molecules (IOI16_molecules)	C++20	0 / 100	0 ms	328 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pll = pair<ll, ll>;

ll n, mn, mx;
vector<pll> v;
vector<int> ans;

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
  n = w.size();
  v.resize(n);
  for (int i = 0; i < n; i++) v[i] = {w[i], i};
  sort(v.begin(), v.end());
  for (int i = 0; i < n; i++) {
    mn += v[i].first;
    mx += v[n-i-1].first;
    if (mn <= u && mx >= l) {
      for (int j = 0; j <= i; j++) ans.push_back(i);
      while (mn < l) {
        auto it = prev(lower_bound(ans.begin(), ans.end(), n-1));
        mn -= v[*it].first;
        (*it)++;
        mn += v[*it].first;
      }
      break;
    }
  }
  for (auto &e: ans) e = v[e].second;
  return ans;
}

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: 0 / 9
• Subtask 2: 0 / 10
• Subtask 3: 0 / 12
• Subtask 4: 0 / 15
• Subtask 5: 0 / 23
• Subtask 6: 0 / 31