LA 3998 Prime k-tuple

题意：如果K个相邻素数p1,p2,p3.....pk满足pk-p1=s，称这些素数组成一个距离为s的素数K元组，输入a,b,k,s，输出区间[a,b]内距离为s的素数k元组的个数。

思路：先打到50000素数表，然后暴力求出a,b区间的素数，然后判断。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int t;
int a,b,k,s,cnt;
bool vis[500000];
int f[500000];
int p[10000000];
void getprime()
{
    cnt=0;
    memset(vis,0,sizeof(vis));
    vis[1]=true;
    vis[0]=true;
    for(int i=2; i<=50000; i++)
    {
        if(!vis[i])
        {
            f[cnt++]=i;
            for(int j=i*2; j<=50000; j+=i)
            {
                vis[j]=true;
            }
        }
    }
}

int main()
{
    getprime();
    scanf("%d
",&t);
    while(t--)
    {
        scanf("%d%d%d%d",&a,&b,&k,&s);
        memset(p,0,sizeof(p));
        int ans=0;
        int t1=0;
        for(int i=a; i<=b; i++)
        {
            if(i<=50000)
            {
                if(!vis[i]) p[t1++]=i;
            }
            else
            {
                bool flag=true;
               for(int j=0; j<cnt&&f[j]*f[j]<=i; j++)
               {
                    if(i%f[j]==0)
                    {
                        flag=false;
                        break;
                    }
               }
               if(flag)
               {
                   p[t1++]=i;
               }
            }
        }
        for(int i=0; i<t1; i++)
        {
            if(p[i+k-1]-p[i]==s)
            {
                ans++;
            }
        }
        printf("%d
",ans);
    }
    return 0;
}