leetcode204. 计数质数

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

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

厄拉多赛筛法,先找到不超过根号n的所有素数,再把这些素数的所有倍数去掉:

class Solution:
    def countPrimes(self, n: int) -> int:
        # 厄拉多赛筛法,先找到不超过根号n的所有素数,把这些素数的所有倍数去掉
        if n <= 2:
            return 0
        res = [1] * n  #  位置i为1表示为素数
        res[:2] = [0]*2
        for i in range(int(n**0.5)+1):
            if res[i] == 1:
                res[i*i:n:i] = [0]*len(res[i*i:n:i])  # 从i*i开始
        return sum(res)