204. 计数质数

统计所有小于非负整数 n 的质数的数量。
示例:
输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

解题思路

埃拉托色尼筛选法，循环中置对应值的倍数为0，最后统计为1的个数，也就是质数的个数。

class Solution:
    def countPrimes(self, n: int) -> int:
        if n<=2:
            return 0 
        # 生成一个n维列表
        isPrime = [1] * n 
        # 0和1不是质数，所以前两个位置为0
        isPrime[0] = isPrime[1] = 0 
        # 此时从index=2开始遍历，isPrime[2]=0，表明第一个质数为2，
        #然后将2的倍数对应的索引全部赋0，以此类推，剩下的索引对应的书就是质数

        # 这里没有用(2, n), 因为求n以内的质数，只要将sqrt(n)以内质数的倍数全部删掉，
        # 剩下的就全是质数
        for i in range(2, int(n**0.5)+1):
            if isPrime[i]==1:
                isPrime[i*i:n:i] = [0] * len(isPrime[i*i:n:i])

        return sum(isPrime)