Submission #310931

#	Time	Username	Problem	Language	Result	Execution time	Memory
310931		MilosMilutinovic	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	62 ms	6252 KiB

molecules

/**
 *  author: milos
 *  created: 08.10.2020 16:11:14
**/
#include <bits/stdc++.h>
#include "molecules.h"

using namespace std;

vector<int> find_subset(int l, int u, vector<int> a) {
  int n = (int) a.size();
  vector<pair<int, int>> v;
  for (int i = 0; i < n; i++) {
    v.push_back({a[i], i});
  }
  sort(v.begin(), v.end());
  long long sum = v[0].first;
  int i = 0, j = 0;
  while (i < n && j < n) {
    if (sum >= l && sum <= u) {
      break;  
    }
    if (sum <= u && j < n - 1) {
      if (j + 1 == n) {
        break;
      }
      j++;
      sum += v[j].first;
    } else {
      sum -= v[i].first;
      i++;
    }
  }
  vector<int> ans;
  if (sum >= l && sum <= u) {
    for (int h = i; h <= j; h++) {
      ans.push_back(v[h].second);
    }  
  }
  sort(ans.begin(), ans.end());
  return ans;
}

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