【洛谷】P3383 【模板】线性筛素数（配数学证明）

题目地址：

https://www.luogu.com.cn/problem/P3383

题目背景：
本题已更新，从判断素数改为了查询第$k$小的素数。提示：如果你使用cin来读入，建议使用 std::ios::sync_with_stdio(0)来加速。

题目描述：
如题，给定一个范围$n$，有$q$个询问，每次输出第$k$小的素数。

输入格式：
第一行包含两个正整数$n,q$，分别表示查询的范围和查询的个数。接下来$q$行每行一个正整数$k$，表示查询第$k$小的素数。

输出格式：
输出$q$行，每行一个正整数表示答案。

数据范围：
对于$100\%$的数据，$n=10^8$，$1\le q\le 10^6$，保证查询的素数不大于$n$。

可以用经典的欧拉筛法在$O(n)$时间内解决（不包含查询所费时间）：

#include <iostream>
using namespace std;

const int N = 1e8 + 10;
int n, q;
// 记录已得到的素数
int p[N], cnt;
// 表示数i是否被筛掉
bool st[N];

void get_primes(int n) {
  // 循环1
  for (int i = 2; i <= n; i++) {
    if (!st[i]) p[++cnt] = i;
    // 循环2
    for (int j = 1; p[j] <= n / i; j++) {
      st[p[j] * i] = true;
      // 此判断条件保证了每个合数只会被它的最小素因子筛掉，不会被筛两遍，从而降低了复杂度
      if (i % p[j] == 0) break;
    }
  }
}

int main() {
  scanf("%d%d", &n, &q);
  get_primes(n);
  while (q--) {
    int x;
    scanf("%d", &x);
    printf("%d\n", p[x]);
  }
}

算法的正确性和复杂度证明：

正确性证明：首先证明，对于循环1，每轮循环结束时，不超过$i$的素数都已经被加入$primes$。用归纳法：当$i=2$时，结论正确。设$i=k-1$时结论正确，当$i=k$，若$k$是素数，那么它不可能被筛数语句筛掉；否则，设$k$的最小素因子为$p$，那么由归纳假设，存在$j$使得$p=primes[j]$，那么当$i=k/p$时，循环里的筛数语句就把$k$筛掉了，证明结束。

复杂度证明：关键在于，要证明筛数语句对于每个合数只筛了一遍。由正确性证明，任何合数$n$，当$primes[j]$是$n$的最小素因子，也即$i=n/primes[j]$的时候，会被筛掉。接下来反证法，如果该合数$n$被筛了两遍，那么存在$s>0$和$i^{\prime }<i$，使得$n=i\ast primes[j]=i^{\prime }\ast primes[j+s]$，由素数的性质知$primes[j]|i^{\prime }$，那么循环1进行到$i^{\prime }$时，循环2还没来得及筛掉$n$的时候已经break跳出循环了，产生了矛盾。所以每个合数只被筛一遍。既然筛数语句总共进行了$O(N)$遍，整个算法复杂度当然就是$O(N)$。

事实上这个算法可以直接求出每个数的最小素因子：

int min_prime_factor[400001];
vector<int> primes;

void inline euler_sieve(int n)
{
    for (int i = 2; i <= n; i++) {
        if (min_prime_factor[i] == 0) primes.push_back(i);
        for (int j = 0; i * primes[j] <= n; j++) {
            min_prime_factor[i * primes[j]] = primes[j];
            if (i % primes[j] == 0) break;
        }
    }

    for (int i = 0; primes[i] <= n; ++i) {
        min_prime_factor[primes[i]] = primes[i];  // 把素数标记出来
    }
}

当然根据最小素因子也能判断该数是不是素数。