质数问题

题目链接

给定两个整数 n 和 k，请你判断在 [2,n] 的范围内是否存在不少于 k 个质数，满足可以表示为两个相邻质数与 1 的和。

例如，19 满足条件，因为 19=7+11+1。

输入格式
第一行包含整数 T，表示共有 T 组测试数据。
每组数据占一行，包含两个整数 n 和 k。

输出格式
每组数据输出占一行，如果存在不少于 k 个质数满足条件则输出 YES，否则输出 NO。

数据范围
1≤T≤30,
2≤n≤1000,
0≤k≤1000

输入样例：

5
27 2
45 7
2 0
15 1
17 1

输出样例：

YES
NO
YES
YES
YES

解题思路：

由于本题数据范围较小，所以我们可以去先枚举出所有的质数，然后再通过遍历的方式去判断我们所要判断的质数能否可以表示为两个相邻质数与 1 的和。

代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>

using namespace std;

int primes[1010],sum;
bool st[1010];

//朴素筛法
void get_primes(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!st[i])
        {
            primes[sum++]=i;
            for(int j=2*i;j<=n;j+=i) st[j]=true;
        }
    }
}

int main()
{
    int t;
    cin>>t;
    get_primes(1010);
    
    while(t--)
    {
        int n,k;
        cin>>n>>k;
        int cnt=0;
        for(int i=2;i<=n;i++)
        {
            if(st[i]) continue;
            for(int j=0;j<sum-1;j++)
            {
                if(primes[j]+primes[j+1]+1==i)
                {
                    cnt++;
                    break;
                }
            }
        }
        
        if(cnt>=k) puts("YES");
        else puts("NO");
    }
    
    return 0;
}