不使用itertools的情况下,不重复的组合。

问题描述 投票:0回答:1

我需要得到一个长度为的迭代数的所有排列组合 n 蛮力算法中。我不想使用 itertools 或任何其他外部包。

我想我可以使用堆算法,但是我的代码只能返回一个重复n!次的排列组合。

def permutations(l):
    n = len(l)
    result = []
    c = n * [0]
    result.append(l)
    i = 0;
    while i < n:
        if c[i] < i:
            if i % 2 == 0:
                tmp = l[0]
                l[0] = l[i]
                l[i] = tmp
            else:
                tmp = l[c[i]]
                l[c[i]] = l[i]
                l[i] = tmp
            result.append(l)
            c[i] += 1
            i = 0
        else:
            c[i] = 0
            i += 1
    return result

我不知道为什么会这样。我也想知道是否有更有效的方法,也许用递归函数。

python permutation
1个回答
1
投票

你可以使用下面的方法,这是没有使用其他内部函数,但它使用了递归。

def permutation(lst):
    if len(lst) == 0:
        return []
    if len(lst) == 1:
        return [lst]

    l = [] # empty list that will store current permutation

    for i in range(len(lst)):
       m = lst[i]

       # Extract lst[i] or m from the list. remainderLst is remaining list
       remainderLst = lst[:i] + lst[i+1:]

       # Generating all permutations where m is first element
       for p in permutation(remainderLst):
           l.append([m] + p)
    return l


data = list('12345')
for p in permutation(data):
    print(p)

1
投票

我会使用@David S的答案. 或者你可以使用这段代码:

def permutate(ls):
    if len(ls) == 0: return []
    elif len(ls) == 1: return [ls]
    else:
        ls1 = []
        for i in range(len(ls)):
            x = ls[i]
            y = ls[:i] + ls[i+1:]
            for p in permutate(y): ls1.append([x]+p)
        return ls1

a = str(input("To permutate? ")).split(" ")
for i in permutate(a): print(" ".join(i))

但它基本上是一样的:D

© www.soinside.com 2019 - 2024. All rights reserved.