📜  Python程序用给定的总和获得K个长度的组

📅  最后修改于: 2022-05-13 01:55:37.260000             🧑  作者: Mango

Python程序用给定的总和获得K个长度的组

给定一个列表,我们的任务是编写一个Python程序来提取所有 K 个长度的子列表,并得出给定的总和。

方法:使用sum() + product() + loop  

在这种情况下,所有可能的长度为 K 的子列表都使用 product() 计算,sum() 用于将子列表的总和与所需的总和进行比较。

Python3
# Python3 code to demonstrate working of
# K length groups with given summation
# Using sum + product()
from itertools import product
  
# initializing list
test_list = [6, 3, 12, 7, 4, 11]
               
# printing original list
print("The original list is : " + str(test_list))
  
# initializing Summation 
N = 21
  
# initializing size 
K = 4
  
# Looping for each product and comparing with required summation
res = []
for sub in product(test_list, repeat = K):
  if sum(sub) == N:
    res.append(sub)
          
# printing result
print("The sublists with of given size and sum : " + str(res))


输出: