Python将递归置换函数转换为迭代

Question

我有一个未知数量的整数变量说可以从 [0,9] 我想遍历这些值的所有排列。

如果变量的数量是恒定的，那么编写嵌套的 for 循环就很容易了。 我想出了一个递归函数来做我想做的事，但很好奇是否有一种方法可以迭代地做到这一点。

def nested(array,index):
    n = len(array)
    for i in range(10):
        array[n-index]=i
        #len(array-1) end of array
        if(index == 1):
            print(array)
            #do something later
        else:
            nested(array,index-1)

#generate all permutations, change n to change size of list.            
n = 4
array = [0]*n
nested(array,len(array))

我尝试使用此处找到的所谓“简单方法”-> http://blog.moertel.com/posts/2013-05-11-recursive-to-iterative.html但我无法让它工作。

Answer 1

正如另一位评论者所提到的，关键是使用堆栈模拟尾递归。

请注意，我append() (array, index) append()的元组append()到stack ，这反映了原始递归解决方案中对递归函数的调用。 在迭代开始时，它会执行stack.pop() ，它模仿递归函数的主体。 递归调用变为stack.append() 。

def nested(array):
    stack = []
    n = len(array)
    stack.append((array.copy(), n))
    while(stack):
        array, index = stack.pop()        
        for i in range(10):
            array[n-index]=i
            #len(array-1) end of array
            if(index == 1):
                print(array)
                #do something later
            else:
                stack.append((array.copy(), index-1)) 

#generate all permutations, change n to change size of list.            
n = 4
array = [0]*n
nested(array)

Answer 2

请参考itertools 。 有一个“排列”类可以完美地解决您的问题。