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

39 lines
833 B
Nim
Raw Permalink Normal View History

2019-09-12 10:33:56 -07: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
proc perm( s: openArray[int], emit: proc(emit:openArray[int]) ) =
var s = @s
if s.len == 0:
emit(s)
return
var rc : proc(np: int)
rc = proc(np: int) =
if np == 1:
emit(s)
return
var
np1 = np - 1
pp = s.len - np1
rc(np1) # recurs prior swaps
for i in countDown(pp, 1):
swap s[i], s[i-1]
rc(np1) # recurs swap
let w = s[0]
s[0..<pp] = s[1..pp]
s[pp] = w
rc(s.len)
var se = @[0, 1, 2, 3] #, 4, 5, 6, 7, 8, 9, 10]
perm(se, proc(seq: openArray[int])=
echo seq
)