【算法与数据结构相关】【LeetCode】【204 计数质数】【Python】

题目:统计所有小于非负整数 的质数的数量。

示例:

输入: 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)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值