我有一个数字列表和一定的总和,我需要计算生成该总和的可能方式

时间:2013-04-16 20:54:27

标签: python

此外,允许重复数字。

我提到了该计划:

def subset_sum_recursive(numbers,target,partial):
    s = sum(partial)

    #check if the partial sum is equals to target
    if s == target: 
        print "sum(%s)=%s"%(partial,target)
    if s >= target:
        return # if we reach the number why bother to continue

    for i in range(len(numbers)):
        n = numbers[i]
        remaining = numbers[i+1:]
        subset_sum_recursive(remaining,target,partial + [n]) 

def subset_sum(numbers,target):
    #we need an intermediate function to start the recursion.
    #the recursion start with an empty list as partial solution.
    subset_sum_recursive(numbers,target,list())

if __name__ == "__main__":
    subset_sum([3,9,8,4,5,7,10],15)

    #Outputs:
    #sum([3, 8, 4])=15
    #sum([3, 5, 7])=15
    #sum([8, 7])=15
    #sum([5, 10])=15

但是我没有把计数变量放在哪里,这太令人困惑了

3 个答案:

答案 0 :(得分:1)

您似乎有一个典型的Counting Coins问题。

您在那里看到的所有片段都应该解决您想要解决的问题(它还包括重复使用相同数字的组合)。我觉得这个维基上的这个python版本很方便,如果很慢:

def changes(amount, coins):
    ways = [0] * (amount + 1)
    ways[0] = 1
    for coin in coins:
        for j in xrange(coin, amount + 1):
            ways[j] += ways[j - coin]
    return ways[amount]

print changes(100, [1, 5, 10, 25])
print changes(100000, [1, 5, 10, 25, 50, 100])

如果您想了解更多信息,请参阅this previous answer以查找类似问题 - 它可以解决问题的可能变体并提供一个非常好的解决方案。

答案 1 :(得分:0)

不确定解决这个问题的规范方法,可能是大师知道的。但我发现我的解决方案非常好:

import itertools
inp = [3,9,8,4,5,7,10]
outp = list()
target = 15
for x in range(2,len(inp)):
    outp.extend([ tsum for tsum in itertools.combinations(inp,x) if sum(tsum) == target ])
outp

UPD 对于小输入,这是非常令人满意的。对于大输入,它不能很好地扩展。 - @gnibbler 如果您有大量输入,请考虑阅读this question

答案 2 :(得分:0)

如果找到一个解决方案,您还可以在main和附加列表中定义列表:

if s == target:
        print "sum(%s)=%s"%(partial,target)
        solutions.append("sum(%s)=%s"%(partial,target))

if __name__ == "__main__":
    solutions=[]
    subset_sum([3,9,8,4,5,7,10],15)
    print len(solutions) # output: 4