Python LeetCode(204.计数质数)

Python LeetCode(204.计数质数)

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

示例:

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

Solution:(直接遍历寻找质数的话时间复杂度太高，超出时间限制。可以建立一个大小为n的布尔类型的list，从0开始，设置0，1为False[即非素数]，2置为True，且将小于n的2的倍数均置为FALSE，之后同理，最终True的个数即为素数的个数。)

class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n <= 1:
            return 0
        
        nums = [None] * n
        nums[0],  nums[1] = False, False
        for i in range(n):
            if nums[i] == None:
                nums[i] = True
                for j in range(i+i, n, i):
                    nums[j] = False
        return sum(nums)

solution = Solution()
print(solution.countPrimes(20))

8