思想:
判断n是否小于2,若是则返回0(由于0和1都不是质数)。定义一个dp存放标志位,若为值1则标志是质数,若为值0则标志不是质数。dp[0]=0和dp[1]=0代表数字0和1都不是质数。循环函数helper的功能是给比n小的2的倍数自己除外(例如4,6,8...),3的倍数,5的倍数,7的倍数等等质数的倍数赋值0(代表他们都不是质数)。当其中一个质数的倍数大于n则结束循环。然后计算dp中1的个数即为质数的量。
class Solution:
def countPrimes(self, n):
"""
:type n: int
:rtype: int
"""
def helper(n, dp):
for i in range(2, n):
if dp[i] == 1:
k = i * i
if k >= n:
return
while k < n:
dp[k] = 0
k += i
if n < 2:
return 0
dp = [1] * n
dp[0] = 0
dp[1] = 0
helper(n, dp)
return sum(dp)