线性筛质数

问题

https://www.acwing.com/problem/content/870/

分析

每一个数判断是否为质数都需要从 2 遍历到 自己，对于每一个数的时间复杂度是 n²，总的时间复杂度是 1 + 2² + 3² + … + n²，最后得到的数量级为 n³，本体的数据最大为 1e6，会TLE，需要使用线性筛质数的方式来进行筛分，时间复杂度基本为 O(n)

线性筛质数的主要思路是通过最小质因子筛合数

代码

#include<iostream>

using namespace std ;

const int N = 1e6 + 10 ;

int n ;
int st[N], primes[N] ;
int cnt ;

void get_primes(int n)
{
    for(int i = 2 ; i <= n ; i ++)
    {
        if(!st[i]) primes[cnt ++] = i ;
        for(int j = 0 ; primes[j] <= n / i ; j ++)
        {
            st[i * primes[j]] = 1 ;
            if(i % primes[j] == 0) break ;
        }
    }
}

int main()
{
    cin >> n ;
   
    get_primes(n) ;
    
    cout << cnt << endl ;
    
    return 0 ;
}

代码解读

保证每次 i * primes[j] 的最小质因子都是 primes[j]，如果 i % primes[j] == 0 了，下一次 primes[j + 1] 就要大于 primes[j]，则此时 primes[j + 1] * i 的最小质因数就是 i 的最小质因数 primes[j] 了，而不是 primes[j + 1]，所以会导致重复筛数，所以当 i % primes[j] == 0 时就要停止循环。

使用最小质因子确实可以保证将每一个合数都筛掉，同时将每一个质数都存下来。相当神奇！