我有一个函数,给定一个数字和一个数字列表,如果该数字可以由列表中的数字组成,则返回该函数,其中数字可以根据需要多次使用:
def canSum(target_sum, numbers, memo = {}):
if target_sum in memo:
return memo[target_sum]
if target_sum == 0:
return True
if target_sum < 0:
return False
for n in numbers:
remainder = target_sum - n
if canSum(remainder, numbers) == True:
memo[target_sum] = True
return True, memo
memo[target_sum] = False
return False,memo
它应该返回True
,并False
取决于是否target_sum
可以通过在列表中添加任何数字任意时间来实现numbers
。
例如canSum(4, [2])
应该返回True
,因为2+2
是 4 等等。但是我不明白我的错误在哪里,以下所有内容都应该返回True
。
canSum(4,[2])
# (False, {2: True, 4: False})
canSum(4,[2,1])
# (False, {2: True, 1: True, 3: True, 4: False})
canSum(4,[1,2])
# (True, {1: True, 2: True, 3: True, 4: True})
canSum(10,[2])
# (False, {2: True, 4: False, 6: False, 8: False, 10: False})
canSum(10,[2,3])
# (False, {2: True, 1: False, 4: False, 3: True, 6: False, 5: True, 8: False, 7: True, 10: False})
另外,是否有区别或需要传递memo
给递归函数调用?这似乎没有任何区别。
if canSum(remainder, numbers) == True: # -> if canSum(remainder, numbers, memo) == True:
memo[target_sum] = True
return True, memo