Python 程序获取具有给定总和的 K 个长度组

pythonserver side programmingprogramming更新于 2025/11/9 19:56:17

当需要获取具有给定总和的 ‘K’ 长度组时,可以使用空列表、‘product’ 方法、‘sum’ 方法和 ‘append’ 方法。

示例

下面是相同的演示

from itertools import product

my_list = [45, 32, 67, 11, 88, 90, 87, 33, 45, 32]
print("列表为: ")
print(my_list)

N = 77
print("N 的值为")
print(N)
K = 2
print("K 的值为 ")
print(K)

my_result = []
for sub in product(my_list, repeat = K):
   if sum(sub) == N:
      my_result.append(sub)

print("结果为: " )
print(my_result)

输出

列表为:
[45, 32, 67, 11, 88, 90, 87, 33, 45, 32]
N 的值为
77
K 的值为
2
结果为:
[(45, 32), (45, 32), (32, 45), (32, 45), (45, 32), (45, 32), (32, 45), (32, 45)]

解释

  • 将所需的包导入到环境中。

  • 定义一个列表并显示在控制台上。

  • 定义 N 和 K 的值并显示在控制台上。

  • 定义一个空列表。

  • 确定列表中元素的乘积,并检查它是否等于 N。

  • 如果是,则将其附加到空列表中。

  • 这将作为输出显示在控制台上。


相关文章


有用资源