Ryhmät

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

constexpr int N = 1 << 17;
int spuu[2 * N];

int hae(int i) {
    return spuu[N + i];
}

void kasvata(int i) {
    i += N;
    while(i > 0) {
        ++spuu[i];
        i >>= 1;
    }
}

int summa(int a, int b, int v = 1, int s = N) {
    if(b <= 0 || a >= s) {
        return 0;
    }
    if(a <= 0 && b >= s) {
        return spuu[v];
    }
    int h = s >> 1;
    return summa(a, b, 2 * v, h) + summa(a - h, b - h, 2 * v + 1, h);
}

struct Tila {
    int p = 0;
    vector<int> koot;
    int velka = 0;
    bool tulos = true;
};

int main() {
    cin.sync_with_stdio(false);
    cin.tie(nullptr);

    int n, t;
    cin >> n >> t;

    vector<pair<int, int>> toiveet(n);
    for(int i = 0; i < n; ++i) {
        cin >> toiveet[i].first >> toiveet[i].second;
    }

    sort(toiveet.begin(), toiveet.end());

    vector<Tila> tilat(t);
    vector<pair<int, int>> kaikki;
    for(int ti = 0; ti < t; ++ti) {
        int m;
        cin >> m;
        vector<int>& koot = tilat[ti].koot;
        koot.resize(m);
        for(int& val : koot) {
            cin >> val;
        }
        sort(koot.begin(), koot.end());

        for(int p = 0; p < m; ++p) {
            kaikki.push_back({koot[p], ti});
        }
    }

    std::sort(kaikki.begin(), kaikki.end());

    int toiveIdx = 0;
    for(auto [koko, ti] : kaikki) {
        Tila& tila = tilat[ti];
        if(!tila.tulos) {
            continue;
        }
        if(tila.p >= (int)tila.koot.size()) throw 5;
        if(tila.koot[tila.p] != koko) throw 5;

        while(toiveIdx < n && toiveet[toiveIdx].first <= koko) {
            kasvata(toiveet[toiveIdx].second);
            ++toiveIdx;
        }

        int montako = koko + tila.velka;
        tila.velka = 0;
        if(summa(koko, N) < montako) {
            tila.tulos = false;
            continue;
        }

        if(tila.p + 1 < (int)tila.koot.size()) {
            int seuraavaKoko = tila.koot[tila.p + 1];
            int loppuvat = 0;
            if(koko < seuraavaKoko) {
                loppuvat = summa(koko, seuraavaKoko);
            }
            if(montako > loppuvat) {
                tila.velka += montako - loppuvat;
            }
        }

        ++tila.p;
    }

    for(int ti = 0; ti < t; ++ti) {
        if(tilat[ti].tulos) {
            cout << "YES\n";
        } else {
            cout << "NO\n";
        }
    }

    return 0;
}

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: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: ACCEPTED