diff --git a/Math/primes.cpp b/Math/primes.cpp index acbe0cf..3420fe1 100644 --- a/Math/primes.cpp +++ b/Math/primes.cpp @@ -1,8 +1,9 @@ -constexpr int sieve_size = 1e7; -bitset bs; +int sieve_size; +bitset<10000001> bs; vector pr; -void sieve() { +void sieve(int size) { + sieve_size = size + 1; bs.set(); bs[0] = bs[1] = 0; for (ll i = 2; i < sieve_size; ++i) if (bs[i]) { for (ll j = i * i; j < sieve_size; j += i) bs[j] = 0;