首页  >  问答  >  正文

算法 - python 给定一个正整数a和一个包含任意个正整数的 列表 b,求所有<=a 的加法组合

例如,10,[1,2,3]

输出类似:
1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1
2 + 2 + 2 +2 + 2
3 + 3 + 3 + 2
3 + 2 + 2 + 2 + 1

注意:是小于等于,list 内的正整数有可能并不能正好等于 a.

PHP中文网PHP中文网2741 天前1166

全部回复(2)我来回复

  • 大家讲道理

    大家讲道理2017-04-18 10:30:42

    通过itertools.combinations_with_replacement我们写短一点的代码:

    def solve2(lst, bound):
        max_length = bound // min(lst)
        for n in range(1, max_length+1):
            for c in itertools.combinations_with_replacement(lst,n):
                if sum(c) <= bound:
                    print('+'.join(map(str, c)))
                
    solve2([1,2,3], 10)

    回复
    0
  • 巴扎黑

    巴扎黑2017-04-18 10:30:42

    假设该问题符合下列假设:

    1. 列表内元素可重复使用

    2. 只要是能满足小于等于上限值的组合都可接受, 就算远小于上限值甚至是零也可以

    以下是暴力法:

    # code for python3
    
    from itertools import combinations
    
    def solve(lst, upperbound):
        candidates = []
        for n in lst:
            for count in range(upperbound//n):
                candidates.append(n)
        allcomb = set()
        for l in range(1, len(candidates)+1):
            for comb in combinations(candidates, l):
                if not comb in allcomb:
                    allcomb.add(comb)
                    if sum(comb) <= upperbound:
                        print('+'.join([str(n)for n in comb]))
            
    solve([1,2,3], 10)

    我回答过的问题: Python-QA

    回复
    0
  • 取消回复