问题
https://www.acwing.com/problem/content/870/
分析
每一个数判断是否为质数都需要从 2 遍历到 自己,对于每一个数的时间复杂度是 n2,总的时间复杂度是 1 + 22 + 32 + … + n2,最后得到的数量级为 n3,本体的数据最大为 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 时就要停止循环。
使用最小质因子确实可以保证将每一个合数都筛掉,同时将每一个质数都存下来。相当神奇!