UVa1404 & LA3998 Prime k-tuple

题目

UVa1404 Prime k-tuple
LA3998 Prime k-tuple

题解

刚开始想标记2*1e9以内的数是不是质数，写了一发RE了，可能开不了那么大的数组嘛qwq。

后来学会一种方法，预处理素数之后暴力找区间内质数（预处理的质数是在check的时候用的）。可以过UVa1401,13350ms,在LA3998上T了，复杂度实在太高。

既然都会筛素数了，筛某个区间内的素数应该不难，仿照埃氏筛法，枚举小于sqrt(b)的所有质数（要求预处理出50000以内质数），然后对于 [a,b]进行筛，枚举到p时，先找到最靠近a的那一个p的倍数x，注意如果x=p的话要+=p，因为p是个质数，埃氏筛法里有一个经典优化是把j的起点换成i*i，这里的x可以与p * p取max，可以加速。然后把素数全找出来了就可以枚举一个端点了，因为长度已知为k所以可以算出另外一个端点。

实测LA3998 502ms UVa1404 300ms qwq很大的改进

代码

//QWsin
#include<cmath>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxv=50000;
const int maxn=50000+10;
const int maxl=200000009;

bool nprime[maxl],not_prime[maxn];
int prime[maxn],pcnt;

inline void init_prime()
{
    int sz=sqrt(maxv);
    for(int i=2;i<=sz;++i) if(!not_prime[i])
        for(int j=i*i;j<=maxv;j+=i) not_prime[j]=1;
    for(int i=2;i<=maxv;++i) if(!not_prime[i]) prime[++pcnt]=i;
}

inline int solve()
{
    int a,b,k,s;cin>>a>>b>>k>>s;
    if(a>b) return 0;

    for(int i=0;i<=b-a;++i) nprime[i]=0;

    int sz=sqrt(b);
    for(int i=1;i<=pcnt&&prime[i]<=sz;++i)
    {
        //找到开头 
        int k=a/prime[i]*prime[i];
        if(k<a) k+=prime[i];
        if(k==prime[i]) k+=prime[i];

        //类似筛法优化 
        k=max(prime[i]*prime[i],k);
        while(k>=a&&k<=b) nprime[k-a]=1,k+=prime[i];
    }

    vector<int>P;
    for(int i=0;i<=b-a;++i) if(!nprime[i]) P.push_back(i+a);

    int ret=0;
    for(int i=0,sz=P.size();i+k-1<sz;++i) if(P[i+k-1]-P[i]==s) ++ret;
    return ret;
}

int main()
{
    int T;cin>>T;init_prime();
    while(T--) printf("%d\n",solve());
    return 0;
}