3792.质数问题

题目描述：
给定两个整数 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

相关代码：

#include <iostream>

using namespace std;

int prime[1001];
int flag = 0;

bool is_prime(int x){ //判断一个数是不是质数
    int flag1 = 1;
    for(int i=2;i<x;i++){
        if(x % i == 0){
            flag1 = 0;
            break;
        }
    }
    if(flag1)
        return true;
    else
        return false;
}

void find_prime(int n){ //找出质数存入数组
    flag = 0;
    for(int i=2;i<=n;i++){
        if(is_prime(i)){
            prime[flag] = i;
            flag++;  //为了知道质数数组的长度
        }
    }
}

int main(){
    int t;
    cin>>t;
    while(t--){
        int n,k;
        cin>>n>>k;
        find_prime(n);
        int cnt = 0;
        for(int i=0;i<flag;i++){ 
            for(int j=0;j<flag-1;j++){
                if(prime[j]+prime[j+1]+1 == prime[i])//判断质数数组中的当前值 是否等于 质数数组中连续两个质数的和 再+1
                    cnt++;
            }
        }
        
        if(cnt >= k)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}