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