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
| const int MAX_N = 1010;
using ll = long long;
bool is_prime[MAX_N];
vector<int> primes;
int init = []() {
for (int i = 0; i < MAX_N; ++i) {
is_prime[i] = true;
}
is_prime[0] = is_prime[1] = false;
for (int i = 2; i < MAX_N; ++i) {
if (!is_prime[i]) {
continue;
}
primes.push_back(i);
for (int j = i; j <= (MAX_N - 1) / i; ++j) {
is_prime[i * j] = false;
}
}
return 0;
} ();
bool check_prime(ll x) {
if (x < MAX_N) {
return is_prime[x];
}
for (ll p : primes) {
if (p * p > x) {
return true;
}
if (x % p == 0) {
return false;
}
}
return true;
}
|