Question

我编写了一个递归函数来获取给定列表的所有可能组合。下面是代码。

它带有一个列表参数并打印所有可能的组合。

def getCombinations(a):
    a2 = []
    ans = ['' for i in range(len(a))]
    helper(a,ans,0)

def helper(a,ans,i):
    if i == len(ans):
        print (ans)
    else:
        ans[i] = ''
        helper(a,ans,i+1)
        ans[i] = a[i]
        helper(a,ans,i+1)

因此，如果我们调用getCombinations([1,2,3])，它将显示如下：

['', '', '']
['', '', 3]
['', 2, '']
['', 2, 3]
[1, '', '']
[1, '', 3]
[1, 2, '']
[1, 2, 3]

我的问题是如何将以上结果存储在列表中。我试图寻找解决方案，但我知道问题是函数从技术上讲不会返回任何内容，但是即使我尝试将return替换为print(...)也不起作用，也不会返回任何内容。

Answer 1

有很多方法可以执行此操作，但是从代码开始，将函数转换为生成器可能是最简单的。

生成器不会一次返回全部结果，而是每个元素可用时。这样，您可以将print替换为yield，并通过将对helper(a,ans,0)的调用包装在list()中来将所有内容收集到一个列表中。

但是，由于代码修改了现有答案，因此您需要收集列表的副本，而不是列表本身，因为这将在以后的迭代中更改。

所以：

from copy import copy

def getCombinations(a):
    ans = ['' for i in range(len(a))]
    return list(helper(a,ans,0))

def helper(a,ans,i):
    if i == len(ans):
        yield copy(ans)
    else:
        ans[i] = ''
        yield from helper(a,ans,i+1)
        ans[i] = a[i]
        yield from helper(a,ans,i+1)

print(getCombinations([1,2,3]))

虽然这是一种非常复杂的方法，但对于空字符串却非常混乱-为什么不使用标准库：

from itertools import combinations
print([c for n in range(4) for c in combinations([1, 2, 3], n)])

或更笼统地说，对于任何列表：

from itertools import combinations
def all_combinations(a):
    return([c for n in range(len(a)+1) for c in combinations(a, n)])
print(all_combinations([1, 2, 3]))

Answer 2

您可以使用从给定列表中提取第一项并将其与列表中其余部分的递归调用返回的每个组合连接起来的函数，直到列表为空：

def getCombinations(lst):
    if lst:
        first, *rest = lst
        for value in '', first:
            for combination in getCombinations(rest):
                yield [value, *combination]
    else:
        yield []

使list(getCombinations([1, 2, 3]))返回：

[['', '', ''],
 ['', '', 3],
 ['', 2, ''],
 ['', 2, 3],
 [1, '', ''],
 [1, '', 3],
 [1, 2, ''],
 [1, 2, 3]]

存储递归函数的输出

2 个答案: