Ryhmät

input/code.cpp: In member function 'void mybits::operator|=(const mybits&)':
input/code.cpp:27:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::array<std::bitset<512>, 196>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |         for (int i = 0; i < data.size(); i++) {
      |                         ~~^~~~~~~~~~~~~
input/code.cpp: In member function 'void mybits::operator^=(const mybits&)':
input/code.cpp:32:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::array<std::bitset<512>, 196>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |         for (int i = 0; i < data.size(); i++) {
      |                         ~~^~~~~~~~~~~~~
input/code.cpp: In member function 'bool mybits::poista(const mybits&, int)':
input/code.cpp:44:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::array<std::bitset<512>, 196>::size_type' {aka 'long unsigned int'} [-Wsign...

#include <iostream>
#include <queue>
#include <vector>
#include <map>
#include <algorithm>
#include <array>
#include <unordered_set>
#include <set>
#include <bitset>

using namespace std;

struct Toive {
    int alku, loppu, indeksi;
};

struct mybits {
    array<bitset<512>, 100001 / 512 + 1> data;
    bool set(int i, bool value) {
        int idx = i / 512;
        int bit = i % 512;
        bool was_set = data[idx][bit];
        data[idx][bit] = value;
        return was_set;
    }
    void operator|=(const mybits& other) {
        for (int i = 0; i < data.size(); i++) {
            data[i] |= other.data[i];
        }
    }
    void operator^=(const mybits& other) {
        for (int i = 0; i < data.size(); i++) {
            data[i] ^= other.data[i];
        }
    }
    int count() const {
        int total = 0;
        for (const auto& b : data) {
            total += b.count();
        }
        return total;
    }
    bool poista(const mybits& other, int määrä) {
        for (int i = 0; i < data.size() && määrä; i++) {
            auto a = data[i] & other.data[i];
            auto c = a.count();
            for (int shl_min = 0, shl_max = 512; c > määrä;) {
                int shl = (shl_min + shl_max) / 2;
                auto a2 = (a << shl) >> shl;
                auto c2 = a2.count();
                if (c2 == määrä) {
                    data[i] ^= a2;
                    return true;
                }
                if (c2 > määrä) {
                    shl_min = shl;
                } else {
                    shl_max = shl;
                }
            }
            määrä -= c;
            data[i] ^= a;
        }
        return määrä == 0;
    }
};

int main() {
    ios::sync_with_stdio(false);
    int n, t;
    cin >> n >> t;
    vector<pair<int, int>> toiveet_loppu_alku(n), toiveet_alku_indeksi(n);
    for (int i = 0; i < n; i++) {
        cin >> toiveet_loppu_alku[i].second >> toiveet_loppu_alku[i].first;
    }
    ranges::sort(toiveet_loppu_alku);
    for (int i = 0; i < n; i++) {
        toiveet_alku_indeksi[i] = {toiveet_loppu_alku[i].second, i};
    }
    ranges::sort(toiveet_alku_indeksi);

    vector<bool> tulokset(t, true);
    vector<array<int, 3>> vaiheet;

    mybits kaikki_vapaana;
    for (auto &a: kaikki_vapaana.data) {
        a.set();
    }

    vector<mybits> vapaat(t, kaikki_vapaana);
    for (int testi = 0; testi < t; testi++) {
        int ryhmien_määrä;
        cin >> ryhmien_määrä;
        map<int, int> ryhmät;
        for (int i = 0; i < ryhmien_määrä; i++) {
            int ryhmän_koko;
            cin >> ryhmän_koko;
            ryhmät[ryhmän_koko] += ryhmän_koko;
        }
        for (auto &[koko, lapsimäärä] : ryhmät) {
            vaiheet.push_back({koko, lapsimäärä, testi});
        }
    }
    ranges::sort(vaiheet);

    auto toive_iter = toiveet_alku_indeksi.begin();
    set<int> toiveet_kesken;
    mybits toiveet_bits;
    for (auto &[koko, lapsimäärä, testi] : vaiheet) {
        if (tulokset[testi] == false) {
            continue;
        }
        while (toive_iter != toiveet_alku_indeksi.end() && toive_iter->first <= koko) {
            int indeksi = toive_iter->second;
            toiveet_kesken.insert(indeksi);
            toiveet_bits.set(indeksi, true);
            toive_iter++;
        }
        while (!toiveet_kesken.empty() && toiveet_loppu_alku[*toiveet_kesken.begin()].first < koko) {
            int indeksi = *toiveet_kesken.begin();
            toiveet_kesken.erase(toiveet_kesken.begin());
            toiveet_bits.set(indeksi, false);
        }
        if (!vapaat[testi].poista(toiveet_bits, lapsimäärä)) {
            tulokset[testi] = false;
        }
    }

    for (auto tulos : tulokset) {
        if (tulos) {
            cout << "YES\n";
        } else {
            cout << "NO\n";
        }
    }
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: RUNTIME ERROR

Test 15

Group: 3

Verdict: RUNTIME ERROR

Test 16

Group: 3

Verdict: ACCEPTED