任何简短的方法来创建固定长度和总和的数组的组合?

Question

例如:

array = [4,0,0]

生成的数组必须是固定长度，数组的和也必须是固定的。获得此输出的最短且有效的方法是什么(允许导入函数):

Output:
[4,0,0]
[0,4,0]
[0,0,4]
[3,1,0]
[0,3,1]
[1,0,3]
[1,3,0]
[0,1,3]
[3,0,1]
[2,1,1]...

Answer 1

下面的解决方案并不完全是您要寻找的，也就是说，它不按照您提到的顺序输出，但是列表内返回的值是相同的。

from itertools import product

def findPairs(sum_value, len_value):
lst = range(sum_value + 1)
return [
pair 
for pair in product(lst, repeat=len_value) 
if sum(pair) == sum_value
]
# ----------------------------------------- #
array = [4, 0, 0]
sum_value = sum(array)
len_value = len(array)
print(findPairs(sum_value, len_value))