Submission #632678

#	Time	Username	Problem	Language	Result	Execution time	Memory
632678		ghostwriter	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	55 ms	6188 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
#include <debug.h>
#endif
#define st first
#define nd second
#define pb push_back
#define pf push_front
#define _pb pop_back
#define _pf pop_front
#define lb lower_bound
#define ub upper_bound
#define mtp make_tuple
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
typedef long long ll; typedef unsigned long long ull;
typedef double db; typedef long double ldb;
typedef pair<int, int> pi; typedef pair<ll, ll> pll;
typedef vector<int> vi; typedef vector<ll> vll; typedef vector<pi> vpi; typedef vector<pll> vpll;
typedef string str;
template<typename T> T gcd(T a, T b) { return (b == 0? a : gcd(b, a % b)); }
template<typename T> T lcm(T a, T b) { return a / gcd(a, b) * b; }
#define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
#define FOS(i, r, l) for (int (i) = (r); (i) >= (l); --(i))
#define EACH(i, x) for (auto &(i) : (x))
#define WHILE while
#define file "TEST"
mt19937 rd(chrono::steady_clock::now().time_since_epoch().count());
ll rand(ll l, ll r) { return uniform_int_distribution<ll>(l, r)(rd); }
/*
    Tran The Bao
    CTL - Da Lat
    Practising for VOI23 gold medal
*/
vi find_subset(int l, int u, vi w) {
    vpi w1;
    int pos = 0;
    EACH(i, w) w1.pb({i, pos++});
    sort(all(w1));
    int cur = 0;
    deque<int> d;
    EACH(i, w1) {
        int wi = i.st, pos = i.nd;
        // debug(1LL * cur + wi);
        if (1LL * cur + wi > u) {
            if (d.empty()) continue;
            cur -= w[d.front()];
            d._pf();
            d.pb(pos);
            cur += wi;
        }
        else {
            d.pb(pos);
            cur += wi;
        }
        if (l <= cur && cur <= u) {
            vi ans;
            EACH(j, d) ans.pb(j);
            return ans;
        }
    }
    return vi(0);
}

Compilation message (stderr)

molecules.cpp: In function 'vi find_subset(int, int, vi)':
molecules.cpp:27:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   27 | #define EACH(i, x) for (auto &(i) : (x))
      |                               ^
molecules.cpp:40:5: note: in expansion of macro 'EACH'
   40 |     EACH(i, w) w1.pb({i, pos++});
      |     ^~~~
molecules.cpp:27:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   27 | #define EACH(i, x) for (auto &(i) : (x))
      |                               ^
molecules.cpp:44:5: note: in expansion of macro 'EACH'
   44 |     EACH(i, w1) {
      |     ^~~~
molecules.cpp:27:31: warning: unnecessary parentheses in declaration of 'j' [-Wparentheses]
   27 | #define EACH(i, x) for (auto &(i) : (x))
      |                               ^
molecules.cpp:60:13: note: in expansion of macro 'EACH'
   60 |             EACH(j, d) ans.pb(j);
      |             ^~~~

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