Algorithm/utils/cal_prime.py

16 lines
298 B
Python
Raw Permalink Normal View History

2022-04-10 05:50:28 +08:00
def cal_prime(n):
prime = [1] * (n + 1)
prime[0] = prime[1] = 0
for i in range(2, n + 1):
j = i * i
while j <= n:
prime[j] = 0
j += i
res = []
for idx, num in enumerate(prime):
if num:
res.append(idx)
return res