Submission #763891

#	Time	Username	Problem	Language	Result	Execution time	Memory
763891		ind1v	Detecting Molecules (IOI16_molecules)	C++11	100 / 100	38 ms	4788 KiB

molecules

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

using namespace std;

vector<int> find_subset(int l, int u, vector<int> w) {
  int n = w.size();
  vector<int> ord(n);
  iota(ord.begin(), ord.end(), 0);
  sort(ord.begin(), ord.end(), [&w](int &i, int &j) -> bool {
    return w[i] < w[j];
  });
  deque<int> dq;
  long long sum = 0;
  for (int i = 0; i < n; i++) {
    if (sum + w[ord[i]] < l) {
      dq.emplace_back(ord[i]);
      sum += w[ord[i]];
    } else if (l <= sum + w[ord[i]] && sum + w[ord[i]] <= u) {
      dq.emplace_back(ord[i]);
      sum += w[ord[i]];
      vector<int> ans(dq.begin(), dq.end());
      return ans;
    } else {
      if (dq.empty()) {
        return vector<int>(0);
      } else {
        sum -= w[dq.front()];
        dq.pop_front();
        sum += w[ord[i]];
        dq.push_back(ord[i]);
        if (l <= sum && sum <= u) {
          vector<int> ans(dq.begin(), dq.end());
          return ans;
        }
      }
    }
  }
  return vector<int>(0);
}

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