求前K个素数的几种方法

现在有一个问题，输入K，求前K个素数或者说求第K个素数。常规的想法，就是从0开始往后遍历，是素数就存起来，直到找到第K个。
我们这里介绍三种筛选素数的方法：埃式筛法、线性筛法、6倍数判别法。

• 埃式筛法

#include<iostream>
#include<cmath>
#include<cstdlib>
using namespace std;
typedef long long ll;
ll m, n;
bool prime[10000000 + 10];
void init()
{
    memset(prime, 1, sizeof(prime));
    prime[0] = 0;
    prime[1] = 0;
    for (ll i = 2; i <= m; i++)
    {
        if (!prime[i]) continue;
        for (ll j = i*i; j <= m; j += i)
            prime[j] = 0;
    }
    return;
}
int main()
{
    cin >> m >> n;
    init();
    int ask = 0;
    while (n--)
    {
        cin >> ask;
        if (prime[ask]) cout << "Yes" << endl;
        else cout << "No" << endl;
    }
    return 0;
}

埃式筛法默认所有数都是素数，0和1不是素数，从2开始，把2的整数倍数都赋值为不是素数，直到我们规定的范围。这样扫一圈可以筛出素数和合数，时间复杂度O(nlognlogn)，空间复杂度O(n)。

• 线性筛法

//#include<bits/stdc++.h>
#include <iostream>

using namespace std;
typedef long long ll;
ll m, n, su[10000000 + 10];
bool isprime[10000000 + 10];
void init()
{
    int cnt = 1;
    memset(isprime, 1, sizeof(isprime));//初始化认为所有数都为素数
    isprime[0] = isprime[1] = 0;//0和1不是素数
    for (long long i = 2; i <= m; i++)
    {
        if (isprime[i])
            su[cnt++] = i;//保存素数i
        for (long long j = 1; j<cnt&&su[j] * i<m; j++)
        {
            isprime[su[j] * i] = 0;//筛掉小于等于i的素数和i的积构成的合数
        }
    }

}
int main()
{
    cin >> m >> n;
    init();
    int ask = 0;
    while (n--)
    {
        cin >> ask;
        if (isprime[ask]) cout << "Yes" << endl;
        else cout << "No" << endl;
    }
    return 0;
}

线性筛法和埃式筛法比较相像，也是初始化所有数都是素数，0和1不是素数。从2开始，是素数就保存，不管是不是都筛掉小于等于i的素数和i的乘积。

• 6倍数判别法

  这种方法是基于一个结论：大于等于5的质数一定和6的倍数相邻。
  证明：令X >= 1，将大于等于5的自然数表示出来：6x-1、6x、6x+1、6x+2、6x+3、6x+4、6x+5、6(x+1)……可以看出，不在6的倍数两侧的数：6x+2、6x+3、6x+4一定不是素数，再把6x除去，然，素数只出现在6x的两侧。但要注意这只是一个必要条件，6x两侧的数并不一定都是素数。但是根据这个规律，我们可以把判断的步长设置为6，大大加快了判断速度。
  这种方法一个数最多需要208ms。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int n,m;
bool su(int a){
    if(a==1) return 0;
    if(a==2||a==3) return 1;
    if(a%6!=1&&a%6!=5) return 0;
    int temp=sqrt(a);
    for(int i=5;i<=temp;i+=6)
    {
        if(a%i==0||a%(i+2)==0) return 0;
    }
    return 1;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x;
        scanf("%d",&x);
        if(su(x)) printf("Yes"),cout<<endl; //是质数
        else printf("No"),cout<<endl;
        x=0; 
    }
    return 0;
}