204. 计数质数

204. 计数质数

原始题目链接：https://leetcode.cn/problems/count-primes/

给定整数 n ，返回 所有小于非负整数 n 的质数的数量 。

示例 1：

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

输入：n = 0
输出：0
示例 3：

输入：n = 1
输出：0

提示：

0 <= n <= 5 * 106

解题思路：

埃氏筛的定义：要得到自然数 n 以内的全部质数，必须把不大于 根号n 的所有质数的倍数剔除，剩下的就是质数。

代码实现：

class Solution:
    def countPrimes(self, n: int) -> int:
        # 最小的质数是2
        if n < 2:
            return 0
        
        # 初始化所有质数的个数
        total = [1] * n
        # 0和1不是质数
        total[0] = total[1] = 0

        # 埃式筛：把不大于根号 n 的所有质数的倍数剔除，即total对应的位置置0
        for i in range(2, int(n ** 0.5) + 1):
            if total[i]:
                total[i * i : n : i] = [0] * ((n - 1 - i * i) // i + 1)
        
        return sum(total)

参考文献：
https://leetcode.cn/problems/count-primes/solution/pythonzui-you-jie-fa-mei-you-zhi-yi-liao-ba-by-bru/