#include <iostream>
#include <vector>
#include <bitset>
using namespace std;
// Rozmiar bitsetu musi być stałą znaną w czasie kompilacji
const int MAX_N = 50001;
int main() {
// Szybkie WE/WY - przy 1 mln zapytań to absolutna konieczność!
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n, m, q;
if (!(cin >> n >> m >> q)) return 0;
// Tworzymy wektor bitsetów na stercie.
// Rezerwujemy n + m + 1 miejsc, aby uniknąć błędów indeksowania.
// Łączna pamięć: ok. 450 000 * 6250 bajtów = ok. 2.8 GB (mieści się w 5 GB)
vector<bitset<MAX_N>> zbiory(n + m + 1);
// 1. Inicjalizacja zbiorów bazowych A1...An
// Optymalizacja: i += j zamiast sprawdzania modulo
for (int j = 1; j <= n; j++) {
for (int i = j; i <= n; i += j) {
zbiory[j].set(i);
}
}
// 2. Wykonywanie m operacji
for (int i = 1; i <= m; i++) {
int typ, x, y;
int aktualny_indeks = n + i;
cin >> typ;
if (typ == 1) { // Suma Ax U Ay
cin >> x >> y;
zbiory[aktualny_indeks] = zbiory[x] | zbiory[y];
}
else if (typ == 2) { // Przecięcie Ax ^ Ay
cin >> x >> y;
zbiory[aktualny_indeks] = zbiory[x] & zbiory[y];
}
else if (typ == 3) { // Negacja A'x
cin >> x;
zbiory[aktualny_indeks] = ~zbiory[x];
}
}
// 3. Odpowiedzi na zapytania
for (int i = 0; i < q; i++) {
int x, v;
cin >> x >> v;
// Sprawdzamy bit na pozycji v
if (zbiory[x].test(v)) {
cout << "TAK\n";
} else {
cout << "NIE\n";
}
}
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | #include <iostream> #include <vector> #include <bitset> using namespace std; // Rozmiar bitsetu musi być stałą znaną w czasie kompilacji const int MAX_N = 50001; int main() { // Szybkie WE/WY - przy 1 mln zapytań to absolutna konieczność! ios_base::sync_with_stdio(false); cin.tie(NULL); int n, m, q; if (!(cin >> n >> m >> q)) return 0; // Tworzymy wektor bitsetów na stercie. // Rezerwujemy n + m + 1 miejsc, aby uniknąć błędów indeksowania. // Łączna pamięć: ok. 450 000 * 6250 bajtów = ok. 2.8 GB (mieści się w 5 GB) vector<bitset<MAX_N>> zbiory(n + m + 1); // 1. Inicjalizacja zbiorów bazowych A1...An // Optymalizacja: i += j zamiast sprawdzania modulo for (int j = 1; j <= n; j++) { for (int i = j; i <= n; i += j) { zbiory[j].set(i); } } // 2. Wykonywanie m operacji for (int i = 1; i <= m; i++) { int typ, x, y; int aktualny_indeks = n + i; cin >> typ; if (typ == 1) { // Suma Ax U Ay cin >> x >> y; zbiory[aktualny_indeks] = zbiory[x] | zbiory[y]; } else if (typ == 2) { // Przecięcie Ax ^ Ay cin >> x >> y; zbiory[aktualny_indeks] = zbiory[x] & zbiory[y]; } else if (typ == 3) { // Negacja A'x cin >> x; zbiory[aktualny_indeks] = ~zbiory[x]; } } // 3. Odpowiedzi na zapytania for (int i = 0; i < q; i++) { int x, v; cin >> x >> v; // Sprawdzamy bit na pozycji v if (zbiory[x].test(v)) { cout << "TAK\n"; } else { cout << "NIE\n"; } } return 0; } |
English