RosettaCodeData/Task/Permutations/Nim/permutations-5.nim

34 lines
702 B
Nim
Raw Permalink Normal View History

2023-07-01 11:58:00 -04:00
# Nim implementation of the (very fast) Go example.
# http://rosettacode.org/wiki/Permutations#Go
# implementing a recursive https://en.wikipedia.org/wiki/SteinhausJohnsonTrotter_algorithm
import algorithm
proc perm(s: openArray[int]; emit: proc(emit: openArray[int])) =
var s = @s
if s.len == 0:
emit(s)
return
proc rc(np: int) =
if np == 1:
emit(s)
return
var
np1 = np - 1
pp = s.len - np1
rc(np1) # Recurse prior swaps.
for i in countDown(pp, 1):
swap s[i], s[i-1]
rc(np1) # Recurse swap.
s.rotateLeft(0..pp, 1)
rc(s.len)
var se = @[0, 1, 2, 3] #, 4, 5, 6, 7, 8, 9, 10]
perm(se, proc(s: openArray[int])= echo s)