Generating permutations with recursive generators
I've been working recently on the Wikipedia article on the Steinhaus–Johnson–Trotter algorithm for listing all the permutations of \( n \) items in such a way that adjacent permutations in the generated sequence differ only by swapping two adjacent elements. Inspired by this, I added an implementation to my PADS library: Permutations.py.
Although the algorithm is most easily explained recursively, most sources that I've seen skip over this quickly in order to get to various non-recursive variants of the algorithm. There are, I think, several reasons for this: the recursive version has too much overhead passing around and copying the permutation, the structure of the recursion makes it more difficult to use the method as part of a larger algorithm since that larger algorithm would have to be split up into a sequence of calls from within the recursive generation procedure, and finally there are variants of the non-recursive algorithm due to Shimon Even and others that make it take constant average time per permutation, something that seems difficult to achieve in a recursive algorithm. But are these reasons actually valid?
The recursive overhead can be eliminated in a very simple way: instead of generating the sequence of permutations recursively, generate the sequence of swaps, and outside of the recursion apply these swaps to the permutations in sequence.
The Python language has the concept of a simple generator. These look like functions containing a special keyword, "yield". But when they are called, instead of running the code within the function, a special iterator object is set up. Then, every time the next() method of the iterator is called, it runs the code of the function from its previous stopping point until the next yield keyword, returning the argument of the yield as the result of the next() method. These things can call each other recursively, and when they do they end up turning the call tree of the recursion upside down, exposing the results of the recursion as a bare sequence that can be used in any context. This eliminates the objection about using recursion in larger algorithms.
Finally, as for efficiency: a recursive algorithm for generating the swap sequence can be defined as a top level subroutine that generates the sequence of swaps for the last element of the input permutation (these swaps cause this element to sweep back and forth across the permutation) together with recursive calls to generate the other swaps. But for a permutation of length \( n, \) the recursive calls happen only once every \( n \) steps. That is, almost all of the time is spent in the outer call. If the recursion is \( n \) steps deep, so that the worst case time to generate a new element of the swap sequence is \( O(n), \) but all but a \( 1/n \) fraction of the elements don't recurse and only take time \( O(1), \) then the average time to generate each swap is \( O(1). \)
So that's what my implementation does: sets up a recursive chain of simple generators for the swap sequence, with each generator in the chain using a constant amount of memory and taking a constant amount of time when it is called. The average number of calls per swap, and therefore the average time to generate each permutation, is \( O(1). \)
It would be interesting, I think, to compare this to the non-recursive constant-average-time methods for generating the same sequence, but doing this fairly would require a lower level language than Python. There are also constant-worst-case-time methods, a result that I don't see how to duplicate using recursion and simple generators, but Sedgewick's 1977 survey of permutation generation algorithms says that they're slower in practice than the average-case methods.
Comments:
2011-10-05T01:43:33Z
The Python code is very nice, thank you. The algorithm looks simple.
>It would be interesting, I think, to compare this to the non-recursive constant-average-time methods for generating the same sequence, but doing this fairly would require a lower level language than Python.<
Do you like the D language? I'm trying to translate this permutation Python code to D.
2011-10-05T01:45:58Z
I don't really know enough about D to like it or not like it.