数论——线性筛素数

在刷了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给筛掉了，所以所有的合数都会被筛掉，进而证明该算法是正确的

综上，我们就证明了该算法的正确