diff --git a/math/prime_numbers.cpp b/math/prime_numbers.cpp index 55d76d30d..32661ee0e 100644 --- a/math/prime_numbers.cpp +++ b/math/prime_numbers.cpp @@ -12,11 +12,12 @@ std::vector primes(int max) { std::vector res; std::vector is_not_prime(max+1, false); - for (int i = 2; i <= max; i++) { + for (size_t i = 2; i <= max; i++) { if (!is_not_prime[i]) res.emplace_back(i); for(int p: res) { - if (i*p > max) break; - is_not_prime[i*p] = true; + size_t k = i*p; + if (k > max) break; + is_not_prime[k] = true; if (i%p == 0) break; } }