leetcode204.计数质数

1.题目描述

统计所有小于非负整数 n 的质数的数量。

示例:

输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

2.解题思路

埃拉托斯特尼筛法，简称埃氏筛或爱氏筛，是一种由希腊数学家埃拉托斯特尼所提出的一种简单检定素数的算法。要得到自然数n以内的全部素数，必须把不大于根号n的所有素数的倍数剔除，剩下的就是素数。这个算法的过程如下图所示：

时间复杂度为o(nloglogn)

 

初始化一个isPrime数组，初始时认为0~n-1之间的数全是质数，从2开始遍历，直到n-1为止，判断当前数是否是质数，2是质数，结果+1，把2的倍数（不超过n）在isPrime中全部标记为非质数，遍历到3时不是质数，直接跳过，遍历到4时，由于是2的倍数，之前被标记为质数过了，遍历到5时，是质数，把5的倍数（不超过n）在isPrime中全部标记为非质数，依次向后遍历

参考：https://www.cnblogs.com/grandyang/p/4462810.html

3.代码实现

class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        # 初始化全是质数
        isPrime=[True for _ in range(n)]
        res=0
        for i in range(2,n):
            # 如果不是质数（之前被标记过）
            if not isPrime[i]:
                continue
            res+=1
            # 开始标记
            j=2
            while i*j<n:
                isPrime[i*j] = False
                j+=1
        return res