(day18) leetcode 204.计数质数

呵呵哈哈耶

于 2024-07-15 22:06:14 发布

阅读量315

点赞数 6

文章标签：算法 leetcode 华为od 数据结构

本文链接：https://blog.csdn.net/m0_54414851/article/details/140417474

版权

描述

给定整数 n ，返回 所有小于非负整数 n 的质数的数量 。

示例 1：

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

示例 2：

输入：n = 0
输出：0

示例 3：

输入：n = 1
输出：0

提示：

0 <= n <= 5 *

超时代码

class Solution:
    def countPrimes(self, n: int) -> int:
        num=0
        for i in range(2,n):
            for j in range(2,int(math.sqrt(i))+1):
                if i%j == 0:
                    break
            else:
                num+=1
        return num

优化后的埃氏筛（
来自Leecod题解powcai)

class Solution:  
    def countPrimes(self, n: int) -> int:  
        # 定义一个函数 countPrimes，接受一个整数 n 作为参数，并返回一个整数
        if n < 2:  
            # 如果 n 小于 2，直接返回 0，因为 2 是最小的素数，n 小于 2 不会有素数
            return 0  
        
        # 创建一个长度为 n 的列表 isPrimes，初始化为全 1，表示所有数都是潜在的素数
        isPrimes = [1] * n  
        
        # 将索引 0 和 1 对应的位置设为 0，因为 0 和 1 不是素数
        isPrimes[0] = isPrimes[1] = 0  
        
        # 遍历从 2 到 sqrt(n) 的整数（包含 sqrt(n)），这是埃拉托色尼筛法的优化步骤
        for i in range(2, int(math.sqrt(n)) + 1):  
            if isPrimes[i] == 1:  
                # 如果 i 是素数（即 isPrimes[i] 为 1）
                
                # 将 i 的所有倍数（从 i*i 开始，到 n 结束，步长为 i）标记为非素数（设为 0）
                isPrimes[i * i:n:i] = [0] * len(isPrimes[i * i:n:i])  
        
        # 返回 isPrimes 列表中所有素数的个数（即值为 1 的元素的个数）
        return sum(isPrimes)

效率提升的关键在于埃拉托斯特尼筛法，简称埃式筛，也叫厄拉多塞筛法：

要得到自然数 n 以内的全部质数，必须把不大于根号n 的所有质数的倍数剔除，剩下的就是质数。

基本思路

埃氏筛的基本思路是：

从2开始，依次标记每个素数的倍数为非素数。
重复上述步骤，直到处理完所有小于等于√n的数。

具体步骤

初始化：
- 创建一个布尔列表 isPrimes，长度为 n，初始化为全 True。isPrimes[i] 表示数字 i 是否是素数。
- 将 isPrimes[0] 和 isPrimes[1] 设为 False，因为0和1不是素数。
标记非素数：
- 从 2 开始遍历，每次找到一个未被标记为非素数的数字 i，将其所有倍数（从 i*i 开始）标记为非素数。
- 之所以从 i*i 开始，是因为所有小于 i*i 的倍数在之前已经被处理过。
优化：
- 遍历的终止条件为 i <= √n，因为如果 i > √n，则 i 的所有倍数都已经在之前被标记。