【lc刷题】204 计数质数_Day47(132/300)

Loewi大湿

于 2019-06-25 23:36:23 发布

阅读量146

点赞数

分类专栏： Leetcode 文章标签： python leetcode 刷题 prime

本文链接：https://blog.csdn.net/weixin_42317507/article/details/93671800

版权

Leetcode 专栏收录该内容

144 篇文章 0 订阅

订阅专栏

132/300

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

示例:

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

思路：厄拉多塞筛法

sieve of Eratosthenes：给素数的倍数灭灯
在这里插入图片描述
图源：wikimedia

Time complexity: O(nloglogn)； Space complexity: O(n)

遍历不需要从 2 ～ N，只需要从 2 ～ $\sqrt{N}$ ：
对于合数N = $\sqrt{N}\times\sqrt{N}$ = d1 $\times$ d2
推出 d1 <= $\sqrt{N}$ or d2 <= $\sqrt{N}$

举例：
119 = 7 $\times$ 17 ； $\sqrt{119}$ = 10.908
7的倍数， 119被灭

关于如何灭灯：需要用到Python 字符串/列表切片slice的知识

class Solution:
    def countPrimes(self, n: int) -> int:
 
        if n < 3: return 0
        
        res = [ 1 for _ in range(n) ]
        
        res[0], res[1] = 0,  0

        for i in range(2, int(n**0.5) + 1): 

            res[i*i::i] = len(res[i*i::i])*[0]
 
        return sum(res)