204. 计数质数

题目

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

示例:

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

解法

class Solution:
    def countPrimes(self, n: int) -> int:
        if n <= 2:
            return 0
        # 筛子
        isPrime = [1]*n
        isPrime[0],isPrime[1] = 0,0
        for i in range(int(math.sqrt(n))+1):
            if isPrime[i] == 1:
                for j in range(i*i, n, i):
                    isPrime[j] = 0

        return sum(isPrime)