diff --git a/bench/algorithm/nsieve/1.py b/bench/algorithm/nsieve/1.py index 3e9efa67..e692c4d8 100644 --- a/bench/algorithm/nsieve/1.py +++ b/bench/algorithm/nsieve/1.py @@ -5,6 +5,9 @@ def nsieve(n): count = 0 flags = [True] * n for i in range(2, n): + if i * i >= n: + count += flags[i:].count(True) + break if flags[i]: count += 1 for j in range(i << 1, n, i):