使用递归创建列表组合

时间:2014-10-01 07:55:58

标签: python recursion

我在列表中创建元素组合时遇到了麻烦。

我想要做的是在Python中创建一个递归函数,它返回一个元素组合,例如list a = [1,2,3,4,5,6,7,8],结果将是组合[1,2,3,4],[1, 3,4,5],[1,4,5,6],[1,2,4,5]等。对于8个元素,它应该返回70个组合(如果我做了我的数学权利)。虽然最好的选择是组合不重复。

我尝试对其进行编码,但我得到的只是[1,2,3,4],[1,3,4,5] etc而不是组合[1,5,7,8]

我知道有一个特殊功能,但我想以递归方式进行。有什么建议吗?

nimed = ["A","B","C","D","E","F","G","H"]


def kombinatsioonid(listike,popitav):
  if len(listike) < 4:
      return
  tyhi = []
  for c in range(len(listike)):
      tyhi.append(listike[c])
  listike.pop(popitav)
  print(tyhi)
  kombinatsioonid(listike,popitav)

kombinatsioonid(nimed,1) 

2 个答案:

答案 0 :(得分:2)

这可以通过这种方式完成:

def combination(l,n, mylist=[]):
    if not n:  print(mylist)
    for i in range(len(l)):
        mylist.append(l[i])
        combination(l[i+1:], n-1, mylist)
        mylist.pop()

l = ["A","B","C","D","E","F","G","H"]
n=4
combination(l, n)

答案 1 :(得分:1)

对于x中的每个元素a,从右边的元素生成所有k-1个组合,并在每个元素前加x。如果是k==0,只需返回一个空组合,从而退出递归:

def combs(a, k):
    if k == 0:
        return [[]]
    r = []
    for i, x in enumerate(a):
        for c in combs(a[i+1:], k - 1):
            r.append([x] + c)
    #print '\t' * k, k, 'of', a, '=', r
    return r

取消注释“打印”行,看看发生了什么。

作为旁注,最好使用英文变量和函数名称,只是为了互操作性(你的问题就是一个例子)。