题目:
统计所有小于非负整数 n 的质数的数量。
示例:
输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。
思路:
- 采用埃拉托色尼筛选法,这个算法的过程如下图所示,我们从2开始遍历到根号n,先找到第一个质数2,然后将其所有的倍数全部标记出来,然后到下一个质数3,标记其所有倍数,一次类推,直到根号n,此时数组中未被标记的数字就是质数。我们需要一个n-1长度的bool型数组来记录每个数字是否被标记,长度为n-1的原因是题目说是小于n的质数个数,并不包括n。
代码:
class Solution:
def countPrimes(self, n: int) -> int:
if n <3:
return 0
else:
flag=[True]*n#小于n的整数一共有n个
flag[0],flag[1]=False,False
# 比如说求9以内的质数个数,那么只要划掉sqrt(9)以内的质数倍数,剩下的即全为质数. 所以在划去倍数的时候也是从i*i开始划掉,而不是i+i.
for i in range(2,int(n**0.5)+1):
if flag[i]:
# 将倍数置从i*i开始,n结束,间隔为i
flag[i*i:n:i]=[False]*len(flag[i*i:n:i])
return sum(flag)
参考原文:https://blog.csdn.net/xy010902100449/article/details/49361263