Ryhmät

input/code.cpp: In function 'int main()':
input/code.cpp:95:18: warning: variable 'toive' set but not used [-Wunused-but-set-variable]
   95 |             auto toive = toiveet_alkaneet.begin()->second;
      |                  ^~~~~

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

using namespace std;

int main() {
    ios::sync_with_stdio(false);
    set<int> rajakohdat;
    int n, t;
    cin >> n >> t;
    vector<array<int, 2>> toiveet(n);
    for (int i = 0; i < n; i++) {
        cin >> toiveet[i][0] >> toiveet[i][1];
        toiveet[i][1] += 1;
        rajakohdat.insert(toiveet[i][0]);
        rajakohdat.insert(toiveet[i][1]);
    }

    set<int> testikohdat;
    vector<map<int, int>> testit(t);
    for (int i = 0; i < t; i++) {
        int m;
        cin >> m;
        for (int j = 0; j < m; j++) {
            int k;
            cin >> k;
            testit[i][k] += k;
            testikohdat.insert(k);
        }
    }

    // Poistetaan rajakohdat, joiden välissä ei ole testikohtaa
    for (auto it = rajakohdat.begin(); it != rajakohdat.end();) {
        auto next_it = next(it);
        if (next_it == rajakohdat.end()) {
            break;
        }
        int alku = *it;
        int loppu = *next_it;
        auto testi_olemassa = testikohdat.lower_bound(alku);
        if (testi_olemassa == testikohdat.end() || *testi_olemassa >= loppu) {
            rajakohdat.erase(it);
        }
        it = next_it;
    }

    map<int, int> muunnokset;
    for (auto r : rajakohdat) {
        muunnokset[r] = muunnokset.size();
    }

    for (auto& toive : toiveet) {
        toive[0] = muunnokset.lower_bound(toive[0])->second;
        toive[1] = muunnokset.lower_bound(toive[1])->second;
    }

    vector<array<int, 3>> vaiheet;
    for (int i = 0; i < t; i++) {
        auto testi = testit[i];
        map<int, int> uusi;
        for (auto [koko, lapsimäärä] : testi) {
            uusi[muunnokset[koko]] += lapsimäärä;
        }
        testi = move(uusi);
        for (auto &[koko, lapsimäärä] : testi) {
            vaiheet.push_back({koko, lapsimäärä, i});
        }
    }
    ranges::sort(vaiheet);

    vector<unordered_set<int>> käytetyt(t);
    vector<bool> tulokset(t, true);
    multimap<int, array<int, 3>> toiveet_tulossa;
    for (int i = 0; i < n; i++) {
        toiveet_tulossa.insert({toiveet[i][0], {toiveet[i][0], toiveet[i][1], i}});
    }
    multimap<int, array<int, 3>> toiveet_alkaneet;
    for (auto &[koko, lapsimäärä, testi] : vaiheet) {
        if (tulokset[testi] == false) {
            continue;
        }
        while (!toiveet_tulossa.empty() && toiveet_tulossa.begin()->first <= koko) {
            auto toive = toiveet_tulossa.begin()->second;
            toiveet_alkaneet.insert({toive[1], toive});
            toiveet_tulossa.erase(toiveet_tulossa.begin());
        }
        while (!toiveet_alkaneet.empty() && toiveet_alkaneet.begin()->first <= koko) {
            auto toive = toiveet_alkaneet.begin()->second;
            toiveet_alkaneet.erase(toiveet_alkaneet.begin());
        }
        for (auto [loppu, toive] : toiveet_alkaneet) {
            if (käytetyt[testi].find(toive[2]) == käytetyt[testi].end()) {
                käytetyt[testi].insert(toive[2]);
                lapsimäärä--;
                if (lapsimäärä == 0) {
                    break;
                }
            }
        }
        if (lapsimäärä > 0) {
            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: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED