3792 质数问题（枚举 + 线性筛）

1. 问题描述：

给定两个整数 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
来源：https://www.acwing.com/problem/content/description/3795/

2. 思路分析：

分析题目可以知道首先我们需要求解出2~n中的所有质数，因为数据规模不是特别大所以我们可以使用线性筛筛选出2~1010的所有质数，然后枚举1~n中的所有数字，如果当前是质数的情况下那么枚举是否存在两个质数的和 + 1 = n，如果相等那么计数加1，最后判断求解出的符合要求的质数个数是否大于等于k即可。

3. 代码如下：

from typing import List


class Solution:
    count = None

    # 线性筛求解2~n的所有质数
    def init(self, n: int, primes: List[int], st: List[int]):
        for i in range(2, n):
            if st[i] == 0:
                primes[self.count] = i
                self.count += 1
            j = 0
            while primes[j] * i < n:
                st[primes[j] * i] = 1
                if i % primes[j] == 0:
                    break
                j += 1

    def process(self):
        N = 1010
        # st[i]标记i是否是一个质数, 等于1不是质数, 等于0是一个质数
        primes, st = [0] * N, [0] * N
        self.count = 0
        self.init(N, primes, st)
        T = int(input())
        for c in range(T):
            n, k = map(int, input().split())
            count = 0
            for i in range(2, n + 1):
                # st[i] = 1的时候表示i不是一个质数跳过
                if st[i] == 1: continue
                # 枚举是否存在相邻的两个质数满足要求
                for j in range(1, self.count):
                    if primes[j] + primes[j - 1] == i - 1:
                        count += 1
                        break
            if count >= k:
                print("YES")
            else:
                print("NO")


if __name__ == '__main__':
    Solution().process()