【每天学习一点算法 2026/01/08】计数质数

每天学习一点算法 2026/01/08

题目:计数质数

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

统计质数数量,什么是质数?就是只能被 1 或 自身整除的数字

  1. 那么我们只需要判断每个数 i 在 [2, i - 1] 范围内是否有能整除 i 的数字,有的话就不是质数,我们先定义一个存放质数的数组,循环判断小于 n 的数字是否是质数,然后将质数放到数组中,最后返回数组长度即可。

    typescript 复制代码
    function countPrimes(n: number): number {
      if (n <= 2) return 0 // 没有小于 2 的质数
      const primes: number[] = []
      a: for (let i = 2; i < n; i++) {
        for (let j = 2; j < i; j++) {
          if (i % j === 0) {
            continue a // 遇到能整除的就直接判断下一个数字
          }
        }
        primes.push(i) // 循环完成都没有跳出说明是质数
      }
      return primes.length
    };

    这个方法遇到大的数字肯定是会超时的,单次检查的时间复杂度是 O (n)

  2. 我们来分析一下,单次检查是否是质数的循环能不能优化一下

    我们很容易就是想到,如果 j 能够整除 i 那是不是, i / j 也能整除 i,那我们只需要校验 j <= i / j 情况下的数字就行了,这样是满足质数条件的循环数就可以大大降低

    typescript 复制代码
    function countPrimes(n: number): number {
      if (n <= 2) return 0 // 没有小于 2 的质数
      const primes: number[] = [] 
      a: for (let i = 2; i < n; i++) {
        for (let j = 2; j <= i / j; j++) {
          if (i % j === 0) {
            continue a
          }
        }
        primes.push(i)
      }
      return primes.length
    };

    提交,后面比较大的数还是会超时。

  3. 埃氏筛:我们要换一个思路了,我们知道如果 i 能被 j 整除 (不是质数)就是表示 i 是 j 的倍数,换而言之只要是 j 的倍数就不可能是质数,那我们创建一个数组,用 1 和 0表示这个位置的代表的数是否是质数,默认将所有数字都标记为 1,从 2 开始循环,首先将2的倍数都标记为 0,那么那么剩余 2 到 2 * 2 之间的 3 为质数,再将 3 的倍数全部标记为 0,以后循环至 n - 1,其实就是找出了小于 n 所有不是质数的数,循环过程中没被标记为 0 的就是质数了。

    typescript 复制代码
    function countPrimes(n: number): number {
      if (n <= 2) return 0 // 没有小于 2 的质数
      const isPrime: number[] = new Array(n).fill(1)
      let count = 0
      for (let i = 2; i < n; i++) {
        if (primes[i]) {
          count++
          for (let j = i + i; j < n; j += i) {
            primes[j] = 0
          }
        }
      }
      return count
    };

    这个方法还可以优化的,我们可以看到 6 既是 2 的倍数又是 3 的倍数,所以里面会有很多重复的判断,上面的取倍数都是从两倍开始取的,但是 [2, i - 1] 的倍数前面都是标记过的,所以我们只需要从第 i 倍开始标记就行了。

    typescript 复制代码
    function countPrimes(n: number): number {
      if (n <= 2) return 0 // 没有小于 2 的质数
      const isPrime: number[] = new Array(n).fill(1)
      let count = 0
      for (let i = 2; i < n; i++) {
        if (primes[i]) {
          count++
          for (let j = i * i; j < n; j += i) {
            primes[j] = 0
          }
        }
      }
      return count
    };
  4. 线性筛: 这就是我们的极限了吗?其实这里面还是有重复标记的情况 比如 2 和 3 都会标记 12,那我们怎么能保证这样的数不被重复标记呢?我们需要保证每个数只被他的最小质因数标记,要怎么做呢?放弃 "遍历质数 → 标记所有倍数",改为 "循环倍数 → 标记已知的质数的倍数"

    • 遍历 2 到 n 的每个数i(不管i是质数还是合数);
    • 用已经找到的质数列表里的质数p,去标记i * p为合数(这里其实标记的时 p 的 i 倍);
    • 核心约束:只有当pi * p的最小质因数时,才标记。

    核心就在于如何保证最小质因数才标记

    假设当前遍历到 6,质数列表里有[2, 3, 5],我们用p遍历质数列表:

    1. i = 6, 质数 p = 2

      • i * p = 12p = 2是 12 的最小质因数 → 可以标记 12 为合数;

      • 此时检查i % p == 0,说明p = 2也是i = 6的最小质因数;

      • 此时必须停止遍历后续质数(比如下一个 p = 3),因为如果继续:

        i * p = 6 * 3=18,但 18 的最小质因数是 2,不是 3 → 如果用 3 标记 18,就违反了 "最小质因数标记" 的规则,会导致 18 后续被 2 重复标记。

    这就相当于我们把倍数放在了最外层的循环,遇到包含不同质因数的倍数时我们只取最小质因数,跳过后续质因数的标记。

    typescript 复制代码
    function countPrimes(n: number): number {
      if (n <= 2) return 0 // 没有小于 2 的质数
      const isPrime: number[] = new Array(n).fill(1)
      const primes: number[] = []
      for (let i = 2; i < n; i++) {
        if (isPrime[i]) {
          primes.push(i)
        }
        for (let j = 0; j < primes.length; j++) {
          if (primes[j] * i > n) break
          isPrime[primes[j] * i] = 0
          if (i % primes[j] === 0) break
        }
      }
      return primes.length
    };

    提交后,评分怎么还不如上面的啊?这对吗?

    题目来源:力扣(LeetCode)

相关推荐
Tansmjs19 小时前
实时数据可视化库
开发语言·c++·算法
WBluuue20 小时前
Codeforces 1075 Div2(ABC1C2D1D2)
c++·算法
2401_8384725120 小时前
C++模拟器开发实践
开发语言·c++·算法
31087487620 小时前
0005.C/C++学习笔记5
c语言·c++·学习
s1hiyu20 小时前
实时控制系统验证
开发语言·c++·算法
daad77720 小时前
V4L2_mipi-csi
算法
缘友一世20 小时前
张量并行和流水线并行原理深入理解与思考
学习·llm·pp·tp
楼田莉子20 小时前
C++现代特性学习:C++14
开发语言·c++·学习·visual studio
阳光九叶草LXGZXJ20 小时前
达梦数据库-学习-50-分区表指定分区清理空洞率(交换分区方式)
linux·运维·数据库·sql·学习
2301_7657031420 小时前
C++代码复杂度控制
开发语言·c++·算法