Submission #1232974

#	Time	Username	Problem	Language	Result	Execution time	Memory
1232974		Lemser	Detecting Molecules (IOI16_molecules)	C++20	100 / 100	45 ms	4128 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx2")
#pragma GCC target("popcnt")
using namespace std;
 
using ll = long long;
using ull = unsigned long long;
using lld = long double;
using ii = pair<int,int>;
using pll = pair<ll, ll>;
 
using vi = vector<int>;
using vll = vector<ll>;
using vii = vector<ii>;
using vpll = vector<pll>;
using vlld = vector<lld>;
 
#define all(x) x.begin(),x.end()
#define lsb(x) x&(-x)
#define gcd(a,b) __gcd(a,b)
#define sz(x) (int)x.size()
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define fls cout.flush()
 
#define fore(i, l, r) for (auto i = l; i < r; i++)
#define fo(i, n) fore (i, 0, n)
#define forex(i, r, l) for (auto i = r-1; i >= l; i--)
#define ffo(i, n) forex (i, n, 0)
 
bool cmin(ll &a, ll b) { if (b < a) { a=b; return 1; } return 0; }
bool cmax(ll &a, ll b) { if (b > a) { a=b; return 1; } return 0; }
 
const ll INF = 1e18;
const int mod = 1e9 + 7, LOG = 20;

vector<int> find_subset(int l, int r, vector<int> w) {
    ll n = w.size(), mnk = 0, mxk = 0;
    vector<array<int, 2>> rd;
    fo (i, n) rd.pb({w[i], i});
    sort(all(w));
    sort(all(rd));
    // cout << n << ' ' << l << ' ' << r << '\n';
    fo (i, n) {
        mnk += w[i]; 
        mxk += w[n-i-1];
        if (l <= mnk && mnk <= r) {
            vector<int> ans;
            fo (j, i+1) ans.pb(rd[j][1]);
            return ans;
        }
        /*
        mnk > r no sirve
        */
        if (mnk < l && l <= mxk) { 
            vector<int> ans;
            fore (j, i+1, n) {
                mnk -= w[j - i - 1];
                mnk += w[j];
                if (l <= mnk && mnk <= r) {
                    fore (x, j-i, j+1) ans.pb(rd[x][1]);
                    break;
                }
            }
            return ans;
        }
    }
    return {};
    
}

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