ACM刷题中高效求素数

我们知道，素数就是除了1和它本身没有其他因子的数，所以我们求素数通常是挨个判断能否整除，然后再优化一下，只判断到sqrt(n)以内就行。但是这种方法也比较慢，所以我利用了一些数学的方法进行求素数。

首先，我们知道，绝大多数的数都有2、3、5等这些的，也就是前面这些因子占得比重较大，因此我就直接构造数，然后去判断，这里我只去除2和3的，所有的数都可以用6n到6n+5来表示，其中，只有6n+1和6n+5有可能是素数，所以只判断这些数就减少了2/3的工作量。

其次，利用位操作来节省存储空间。int型的每一位表示第i个数是否为素数。代码如下：

#include <iostream>
#include <stdio.h>
#define N 32620
#define Size (N/6*2 + (N%6 == 5? 2: (N%6>0)))
using namespace std;


int prime[3501];//提前算出32620以内有3501个素数
void all_prime_number();//寻找所有素数，将其存到prime数组中
int main()
{
    all_prime_number();


    return 0;
}


void all_prime_number()
{
    int m,n,j,len,stp,k;//int型每一位表示是否为素数
    bool P[Size]={false};
    int gab = 4;


    for (k = 5; k * k <= N; k += gab ^= 6)
    {
        m=k/6*2 + ((k%6)>>2);
        n=m>>1;
        if(m%2)//6n+5
        {
            for(j=12*n*n+20*n+8,len=4*n+3,stp=12*n+10;(j+len)<Size ;j+=stp)
            {
                P[j] = 1;
                P[j+len] = 1;
            }
            if(j<Size)
                P[j] = 1;
        }
        else//6n+1
        {


            for(j=12*n*n+4*n,len=8*n+1,stp=12*n+2;(j+len)<Size;j+=stp)
            {
                P[j] = 1;
                P[j+len] = 1;
            }
            if(j<Size)
                P[j] = 1;
        }
        //cout<<k<<" ";
    }
    //cout<<(P[0] & 4095)<<endl;
    n=1;
    for(j=0;j<Size;j++)
    {


        if(!P[j])
        {
            prime[n]=j/2*6+((j%2)<<2)+1;
            //cout<<prime[n]<<endl;
            n++;
        }
    }
    prime[0]=2;
    prime[1]=3;
}