P1036 [NOIP2002 普及组] 选数 Python题解

mengsi55

于 2024-09-06 17:32:17 发布

阅读量845

点赞数 29

文章标签： python 开发语言算法 dfs

本文链接：https://blog.csdn.net/weixin_52740785/article/details/141962533

版权

[NOIP2002 普及组] 选数

题目描述

已知 $n$ 个整数 $x_1,x_2,\cdots,x_n$ ，以及 $1$ 个整数 $k$ （ $k < n$ ）。从 $n$ 个整数中任选 $k$ 个整数相加，可分别得到一系列的和。例如当 $n = 4$ ， $k = 3$ ， $4$ 个整数分别为 $3, 7, 12, 19$ 时，可得全部的组合与它们的和为：

$3 + 7 + 12 = 22$

$3 + 7 + 19 = 29$

$7 + 12 + 19 = 38$

$3 + 12 + 19 = 34$

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数： $3 + 7 + 19 = 29$ 。

输入格式

第一行两个空格隔开的整数 $n, k$ （ $\le n \le 20$ ， $k < n$ ）。

第二行 $n$ 个整数，分别为 $x_1,x_2,\cdots,x_n$ （ $\le x_i \le 5\times 10^6$ ）。

输出格式

输出一个整数，表示种类数。

样例 #1

样例输入 #1

4 3
3 7 12 19

样例输出 #1

提示

【题目来源】

NOIP 2002 普及组第二题

题解

本质是一个寻找组合个数的问题，这里给出两种方案，一个是使用Python自带的库解决这个问题，一个是使用深度优先搜索解决这个问题：
质数判断用的是[超详细]3种方法判断一个数是否为质数(Python)

自带库

就是itertools的combinations工具

import math
def is_zhishu(t):
    if t <= 1 or t == 4:
        return False
    elif t in (2, 3):
        return True
    elif t % 6 != 1 and t % 6 != 5:
        return False
    sqrt_t = math.ceil(t ** 0.5)
    for i in range(5, sqrt_t, 6):
        if t % i == 0 or t % (i + 2) == 0:
            return False
    return True

def Solution():
    N, K = map(int, input().split())
    x = list(map(int, input().split()))
    from itertools import combinations
    comb = list(combinations(x, K))
    ans = 0
    for i in comb:
        if is_zhishu(sum(i)):
            ans += 1
    return ans
print(Solution())

深度优先搜索

我们先举个例子：

3 7 12 19

我们想要找出这串数字所有的组合（2个组合），可以使用到深度优先搜索，下面是图：
在这里插入图片描述

import math
def is_zhishu(t):
    if t <= 1 or t == 4:
        return False
    elif t in (2, 3):
        return True
    elif t % 6 != 1 and t % 6 != 5:
        return False
    sqrt_t = math.ceil(t ** 0.5)
    for i in range(5, sqrt_t, 6):
        if t % i == 0 or t % (i + 2) == 0:
            return False
    return True



class Solution2:
    def __init__(self):
        self.ans = 0
    def dfs(self, idx, total, sums):
        if total == self.K:
            if is_zhishu(sums):
                self.ans+=1
                return
        if idx < self.N:
            # 不选
            self.dfs(idx+1, total, sums)
            # 选
            self.dfs(idx+1, total+1, sums+self.x[idx])
        return

    def Solution(self):
        self.N, self.K = map(int, input().split())
        self.x = list(map(int, input().split()))
        self.dfs(0, 0, 0)

kk = Solution2()
kk.Solution()
print(kk.ans)