题目:统计所有小于非负整数 n 的质数的数量。
示例:
输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。
思路:1、从数字2开始,将所有其整数倍从小于n的数中排除,然后依次排除3的倍数、(4已经被排除)、5的倍数....最终剩下的就是小于n的所有质数
代码:
class Solution(object):
def countPrimes(self, n):
"""
:type n: int
:rtype: int
"""
if n < 2:
return 0
else:
result = [i for i in range(2,n)]
for item in result:
k = 2
while item*k < n:
if item*k in result:
result.remove(item*k)
k += 1
return len(result)
运行结果:超出时间限制!!!这么优秀的算法还超出时间限制!!!
看了别人的代码,思路是没问题的,但是实现方式不一样:我是通过list.remove来移除元素,而remove操作较为耗时,参考别人的做法:维护一个数组,数组中的元素为1或0来表示当前位置的索引是否为素数。
class Solution(object):
def countPrimes(self, n):
"""
:type n: int
:rtype: int
"""
if n < 2:
return 0
else:
result = [1]*n
result[0] = result[1] = 0
for i in range(n):
if result[i]:
result[2*i:n:i] = [0]*len(result[2*i:n:i])
return sum(result)