Submission #287255

#	Time	Username	Problem	Language	Result	Execution time	Memory
287255		Tangent	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	68 ms	6256 KiB

molecules

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

//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
// typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;
// find_by_order(), order_of_key()

typedef long long ll;
typedef long double dd;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<dd, dd> pdd;
typedef vector<int> vii;
typedef vector<ll> vll;
typedef vector<dd> vdd;
typedef vector<pii> vpii;
typedef vector<pll> vpll;
typedef vector<pdd> vpdd;
typedef vector<vii> vvii;
typedef vector<vll> vvll;
typedef vector<vdd> vvdd;
typedef vector<vpii> vvpii;
typedef vector<vpll> vvpll;
typedef vector<vpdd> vvpdd;
typedef vector<bool> vb;

const int inf = 1 << 30;

#define rep(i, n) for (ll i = 0; i < n; i++)
#define ffor(i, a, b) for(ll i = a; i < b; i++)
#define forin(x, a) for (auto &x: a)
#define all(x) x.begin(), x.end()

#ifdef TEST
#define dbg(x) cout << #x << ": " << x << '\n';
#define dbgc(x) cout << #x << ":"; forin(a, x) { cout << " " << a; } cout << endl;
#define tassert(x) assert(x);
#else
#define dbg(x)
#define dbgc(x)
#define tassert(x)
#endif

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
  vpii nums;
  rep(i, w.size()) {
    nums.emplace_back(w[i], i);
  }
  sort(all(nums));
  int a = 0, b = 0;
  ll tot = 0;
  while (true) {
    if (tot < l) {
      if (b == w.size()) {
        break;
      }
      tot += nums[b++].first;
    } else if (tot > u) {
      tot -= nums[a++].first;
    } else {
      break;
    }
  }
  vii res;
  if (l <= tot && tot <= u) {
    ffor(i, a, b) {
      res.emplace_back(nums[i].second);
    }
  }
  return res;
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:32:36: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 | #define rep(i, n) for (ll i = 0; i < n; i++)
......
   49 |   rep(i, w.size()) {
      |       ~~~~~~~~~~~                   
molecules.cpp:49:3: note: in expansion of macro 'rep'
   49 |   rep(i, w.size()) {
      |   ^~~
molecules.cpp:57:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |       if (b == w.size()) {
      |           ~~^~~~~~~~~~~

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