给定的数字列表，检查是否存在添加到100的数字。将所有这些数字列出为列表列表

Question

Input : L=[1,50,16,34,8,42,99]

Output : [[50,8,42],[16,34,50],[99,1],[16,34,42,8]]

创建一个函数check_sum_100（l）

我能够找到当元素形成子阵列时总计100的元素的总和。但是我找不到[99,1]之类的元素列表。

这是我尝试实现的方法：

def check_sum_100(L):
    out=[]
    for i in range(0, len(L)):             
        for j in range(i, len(L)): 
            sum = 0 
            temp = []                       
            for k in range (i, j+1):
                sum = sum + L[k]            
                temp.append(L[k])
            if sum == 100:
                out.append(temp)
    return out


L = [1,50,16,34,8,42,99]

o = check_sum_100(L)

print(o)

Answer 1

使用itertools.combinations获取不同长度的sublists并检查总和：

import itertools

def check_sum_100(L):
    output = list()
    for l in range(len(L)+1):
        for sublist in itertools.combinations(L, l):
            if sum(sublist)==100:
                output.append(sublist)         
    return output

>>> check_sum([1,50,16,34,8,42,99])
[(1, 99), (50, 16, 34), (50, 8, 42), (16, 34, 8, 42)]

Answer 2

您可以将递归用于此类问题：

def check_sum_100(L):
    # inner function looks for a specific sum target and generates combinations
    def check_sum(L,target):
        for i,v in enumerate(L):
            if v == target:       # base condition, a number in L is the target
                yield [v]
            else:
                # check the sublist after this number for sums that equal the remainder
                for o in check_sum(L[i + 1:], target - v):
                    yield [v] + o
    return list(check_sum(L, 100))

L = [1,50,16,34,8,42,99]
print(check_sum_100(L))

输出：

[[1, 99], [50, 16, 34], [50, 8, 42], [16, 34, 8, 42]]