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