问题
https://www.acwing.com/problem/content/870/
分析
每一个数判断是否为质数都需要从 2 遍历到 自己,对于每一个数的时间复杂度是 n2,总的时间复杂度是 1 + 22 + 32 + … + n2,最后得到的数量级为 n3,本体的数据最大为 1e6,会TLE,需要使用线性筛质数的方式来进行筛分,时间复杂度基本为 O(n)
线性筛质数的主要思路是通过最小质因子筛合数
线性筛质数
保证每次 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 时就要停止循环。
对于一个数 c = a * b ( b 为 c 的最小质因数),当通过该算法的循环循环至 c * b 时,易得此时 c % b == 0 ,如果此时继续循环至 b 后面的一个素数 d ,则有:c * d = a * b * d = ( a * d ) * b,因为 d > b ,所以 a * d > c 。当循环从 c 继续查找到 a * d 时我们发现当 a * d再次与素数 b 想乘时,就又对 c * d 进行了一次操作,出现了冗余,所以在if(n % prime[j] == 0)成立时要将该层循环break掉;
举个例子,对于一个数 9 ,9 * 2 = 18 将 18 标记为合数,循环继续;9 * 3 = 27 将 27 标记为合数,此时发现 9 % 3 = 0,循环退出。如果将循环继续下去会出现筛除 9 * 5 = 45 的情况,而 45 = 15 * 3,在 15 时会被在筛去一次,故不可行
代码
#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 ;
}