在Python中实现堆的置换算法

Question

我试图用python实现Heap的算法，但在重复一些解决方案时遇到了麻烦。我被臭虫的位置难住了。以下是实施方案：

import copy

def _permute(l, n):
    if n == 1:
        print(l)
    else:
        for i in range(n - 1):
            # note: must copy here, or all answers will be same
            new_arr = copy.copy(l)
            _permute(new_arr, n - 1)

            if n % 2 == 0:
                l[i], l[n - 1] = l[n - 1], l[i]
            else:
                l[0], l[n - 1] = l[n - 1], l[0]

        _permute(l, n - 1)

对于输入[0, 1, 2]和3，我得到：

[0, 1, 2]
[1, 0, 2]
[2, 1, 0]
[1, 2, 0]
[0, 1, 2] ** repeats from first answer **
[1, 0, 2] ** repeats from second answer **

对于从第一个和第二个重复的最后2个结果，缺少：

[0, 2, 1]
[2, 0, 1]

我已经搜索了多个地方并尝试了该算法的不同实现，但无论我尝试了多少次，我似乎都无法让它工作。我错过了什么

Answer 1

您将递归调用放置在错误的位置，这应该可以解决问题（并且您不应该使用副本）：

def _permute(l, n):
    if n == 1:
        print(l)
    else:
        _permute(l, n - 1)
        for i in range(n - 1):
            if n % 2 == 0:
                l[i], l[n - 1] = l[n - 1], l[i]
            else:
                l[0], l[n - 1] = l[n - 1], l[0]

            _permute(l, n - 1)