mirror of
https://github.com/nim-lang/Nim.git
synced 2026-02-14 23:33:28 +00:00
Add nextPermutation and prevPermutation
Fits best into algorithm module I guess. These are the most general ways, an iterator could easily be implemented from this. Same algorithm as in Rust: http://web.mit.edu/rust-lang_v0.11/doc/src/collections/var/tmp/alexp/rust/rust-0.11.0/src/libcollections/slice.rs.html#644
This commit is contained in:
@@ -220,3 +220,62 @@ proc product*[T](x: openArray[seq[T]]): seq[seq[T]] =
|
||||
result.add(res)
|
||||
index = 0
|
||||
indexes[index] -=1
|
||||
|
||||
proc nextPermutation*[T](x: var openarray[T]): bool {.discardable.} =
|
||||
## Calculates the next lexicographic permutation, directly modifying ``x``.
|
||||
## The result is whether a permutation happened, otherwise we have reached
|
||||
## the last-ordered permutation.
|
||||
##
|
||||
## .. code-block:: nim
|
||||
##
|
||||
## var v = @[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
|
||||
## v.nextPermutation()
|
||||
## echo v
|
||||
if x.len < 2:
|
||||
return false
|
||||
|
||||
var i = x.high
|
||||
while i > 0 and x[i-1] >= x[i]:
|
||||
dec i
|
||||
|
||||
if i == 0:
|
||||
return false
|
||||
|
||||
var j = x.high
|
||||
while j >= i and x[j] <= x[i-1]:
|
||||
dec j
|
||||
|
||||
swap x[j], x[i-1]
|
||||
x.reverse(i, x.high)
|
||||
|
||||
result = true
|
||||
|
||||
proc prevPermutation*[T](x: var openarray[T]): bool {.discardable.} =
|
||||
## Calculates the previous lexicographic permutation, directly modifying
|
||||
## ``x``. The result is whether a permutation happened, otherwise we have
|
||||
## reached the first-ordered permutation.
|
||||
##
|
||||
## .. code-block:: nim
|
||||
##
|
||||
## var v = @[0, 1, 2, 3, 4, 5, 6, 7, 9, 8]
|
||||
## v.prevPermutation()
|
||||
## echo v
|
||||
if x.len < 2:
|
||||
return false
|
||||
|
||||
var i = x.high
|
||||
while i > 0 and x[i-1] <= x[i]:
|
||||
dec i
|
||||
|
||||
if i == 0:
|
||||
return false
|
||||
|
||||
x.reverse(i, x.high)
|
||||
|
||||
var j = x.high
|
||||
while j >= i and x[j-1] < x[i-1]:
|
||||
dec j
|
||||
|
||||
swap x[i-1], x[j]
|
||||
|
||||
result = true
|
||||
|
||||
17
tests/stdlib/tpermutations.nim
Normal file
17
tests/stdlib/tpermutations.nim
Normal file
@@ -0,0 +1,17 @@
|
||||
discard """
|
||||
output: '''@[0, 1, 2, 3, 4, 5, 6, 7, 9, 8]
|
||||
@[0, 1, 2, 3, 4, 5, 6, 8, 7, 9]
|
||||
@[0, 1, 2, 3, 4, 5, 6, 8, 9, 7]
|
||||
@[0, 1, 2, 3, 4, 5, 6, 8, 7, 9]
|
||||
@[0, 1, 2, 3, 4, 5, 6, 7, 9, 8]
|
||||
@[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]'''
|
||||
"""
|
||||
import algorithm
|
||||
|
||||
var v = @[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
|
||||
for i in 1..3:
|
||||
v.nextPermutation()
|
||||
echo v
|
||||
for i in 1..3:
|
||||
v.prevPermutation()
|
||||
echo v
|
||||
Reference in New Issue
Block a user