时间戳

首页 > 解决方案 > 函数 if True 返回组合列表

python - 函数 if True 返回组合列表

问题描述

我正在关注动态编程视频。但是，对于下面显示的给定数字集，返回为 False。它应该显示 [3,4]

如果我将“组合”更改为 True。它将返回 True。但是我想展示这个组合。

combo = list()

def howsum(targetsum,numbers):
    if targetsum == 0: return combo
    if targetsum < 0: return False
    
    for number in numbers:
        remainder = targetsum - number
        combo.append(number)
        
        if howsum(remainder,numbers) == True: return combo
    return False
     
print(howsum(7,[3,4])) #output should be [3,4]

标签： pythonmemoization

解决方案

这里的问题是您正在检查返回值是否为 True，而不是它是否存在。空集合的真实性是False，非空集合的真实性是True。同样对于回溯，您需要pop()从当前组合中尝试替代方案。

combo = list()
def howsum(targetsum, numbers):
    if targetsum == 0: return combo
    if targetsum < 0: return False

    for number in numbers:
        remainder = targetsum - number
        combo.append(number)
        if howsum(remainder, numbers): return combo
        combo.pop()
    return False


print(howsum(7, [3, 4]))  # output should be [3,4]

还有一件事要注意。combo我们可以通过引入辅助函数将其作为参数传递，而不是创建全局列表。

def howsum_helper(targetsum, numbers, combo):
    if targetsum == 0: return combo
    if targetsum < 0: return False

    for number in numbers:
        remainder = targetsum - number
        combo.append(number)
        if howsum_helper(remainder, numbers, combo): return combo
        combo.pop()
    return False

def howsum(targetsum, numbers):
    return howsum_helper(targetsum, numbers, [])

print(howsum(7, [3, 4]))  # output should be [3,4]

此解决方案还假设可以重复数字列表中的数字以达到targetSum.

推荐阅读