Ryhmät

input/code.cpp: In function 'int main()':
input/code.cpp:54:37: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   54 |         if (toiveet_alkaneet.size() >= koko) {
      |             ~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <set>
#include <unordered_map>
#include <unordered_set>
 
using namespace std;

typedef pair<int, int> P;

int main() {
    iostream::sync_with_stdio(false);
    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<vector<int>> testit(t);
    set<int> ryhmäkoot;
    for (auto &testi : testit) {
        int ryhmien_maara;
        cin >> ryhmien_maara;
        testi.resize(ryhmien_maara);
        for (int i = 0; i < ryhmien_maara; i++) {
            cin >> testi[i];
            ryhmäkoot.insert(testi[i]);
        }
        sort(testi.begin(), testi.end());
    }

    unordered_map<int, vector<int>> toiveet_ryhmässä(n + 1);
    auto vertaa_indeksilla = [&](int i, int j) {
        return toiveet[i].second < toiveet[j].second;
    };
    vector<int> toiveet_alkaneet;
    int toive_i = 0;
    for (int koko: ryhmäkoot) {
        while (!toiveet_alkaneet.empty() && toiveet[toiveet_alkaneet[0]].second < koko) {
            pop_heap(toiveet_alkaneet.begin(), toiveet_alkaneet.end(), vertaa_indeksilla);
            toiveet_alkaneet.pop_back();
        }
        while (toive_i < n && toiveet[toive_i].first <= koko) {
            if (toiveet[toive_i].second >= koko) {
                toiveet_alkaneet.push_back(toive_i);
                push_heap(toiveet_alkaneet.begin(), toiveet_alkaneet.end(), vertaa_indeksilla);
            }
            toive_i++;
        }
        if (toiveet_alkaneet.size() >= koko) {
            toiveet_ryhmässä[koko] = toiveet_alkaneet;
            sort_heap(toiveet_ryhmässä[koko].begin(), toiveet_ryhmässä[koko].end(), vertaa_indeksilla);
        }
    }

    for (int _ = 0; _ < t; _++) {
        if (0) {
            ei:;
            cout << "NO\n";
            continue;
        }
        vector<int> ryhmät = testit[_];
        unordered_set<int> käytetyt;
        int prev_koko = 0;
        auto toiveet_iter = toiveet_ryhmässä[0].begin();
        for (int koko : ryhmät) {
            if (prev_koko != koko) {
                prev_koko = koko;
                toiveet_iter = toiveet_ryhmässä[koko].begin();
            }
            for (int lapsi = 0; lapsi < koko; lapsi++) {
                while (toiveet_iter != toiveet_ryhmässä[koko].end() && käytetyt.count(*toiveet_iter)) {
                    toiveet_iter++;
                }
                if (toiveet_iter == toiveet_ryhmässä[koko].end()) {
                    goto ei;
                }
                käytetyt.insert(*toiveet_iter);
                toiveet_iter++;
            }
        }
        cout << "YES\n";
    }
}

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

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