算法 - python 給定一個(gè)正整數(shù)a和一個(gè)包含任意個(gè)正整數(shù)的 列表 b,求所有<=a 的加法組合
問(wèn)題描述
例如,10,[1,2,3]
輸出類似:1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 1 + 12 + 2 + 2 +2 + 23 + 3 + 3 + 23 + 2 + 2 + 2 + 1
注意:是小于等于,list 內(nèi)的正整數(shù)有可能并不能正好等于 a.
問(wèn)題解答
回答1:通過(guò)itertools.combinations_with_replacement我們寫(xiě)短一點(diǎn)的代碼:
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)回答2:
假設(shè)該問(wèn)題符合下列假設(shè):
列表內(nèi)元素可重複使用
只要是能滿足小於等於上限值的組合都可接受, 就算遠(yuǎn)小於上限值甚至是零也可以
以下是暴力法:
# code for python3from itertools import combinationsdef 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)
我回答過(guò)的問(wèn)題: Python-QA
