在刷了n次y总的视频后,终于真正掌握了该算法,或许已经可以靠理解写出板子了hhh
话不多说,直接上代码
int primes[N], cnt; // primes[]存储所有素数
bool st[N]; // st[x]存储x是否被筛掉
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[primes[j] * i] = true;
if (i % primes[j] == 0) break;
}
}
}
作者:yxc
链接:https://www.acwing.com/blog/content/406/
来源:AcWing
该算法的时间复杂度看似是两层循环,但实际时间复杂度是O(n)的,毕竟是线性筛法
功能:求出并保存1~n的素数,素数保存在primes[]中,从小到大排序,其次数组st用于标记被筛掉的合数,当枚举到i时,如果st[i] = true,就表明i是合数,反之则i为素数(会不会存在i是合数但是没被筛掉呢?不会,证明在下)
下面证明该算法的正确性和高效性(时间复杂度O(n))
高效性证明(时间复杂度证明)
每个合数只会被筛一次,且一定会被自己的最小质因子筛掉
来到第二层循环
首先就是一个st[primes[j]*i] = true
相当于筛掉了合数primes[j]*i
下面有个条件判断,所以每次循环都会有如下两种情况:
if (i % primes[j] == 0) break;
当是这种情况时,此时primes[j]一定是i的最小质因数,因为primes[j]是从小到大枚举的,所以合数i*primes[j]的最小质因子也一定是primes[j],所以合数i*primes[j]一定是用primes[j]来筛,正好对应前面的语句st[primes[j]*i] = true
if (i % primes[j] != 0)
此时primes[j]一定小于i的最小质因子,因为从小到大枚举的素数,所以合数i*primes[j]最小质因子也一定是primes[j],所以也一定是用primes[j]来筛,对应前面的st[primes[j]*i] = true
综上可知,每个合数一定只会被最小的质因子筛掉,如24只会被2筛掉,而不会被3筛掉,因为每个合数只会被最小质因子筛掉,所以该算法时间复杂度为O(n)
正确性证明
所有的合数都会被筛掉
对于任何一个合数x<n,它一定有最小的质因数primes[j],所以可以写成primes[j]*y(满足primes[j]<y),显然在i = y时,第二层循环会枚举到primes[j],(如果枚举不到primes[j],那就说明存在更小的质数z是y的质因数,进而z一定是x的质因数,这与开始的假设矛盾)则肯定会执行st[primes[j]*i] = true,这样就把合数x给筛掉了,所以所有的合数都会被筛掉,进而证明该算法是正确的