mirror of
https://github.com/aicodix/code.git
synced 2026-04-27 14:30:36 +00:00
split reversible Fisher Yates shuffle into two
This commit is contained in:
parent
8370e6a51c
commit
8a055ac2b2
1 changed files with 15 additions and 9 deletions
24
permute.hh
24
permute.hh
|
|
@ -11,24 +11,30 @@ Copyright 2023 Ahmet Inan <inan@aicodix.de>
|
||||||
namespace CODE {
|
namespace CODE {
|
||||||
|
|
||||||
template <int SIZE>
|
template <int SIZE>
|
||||||
class ReversibleFisherYatesShuffle
|
struct FisherYatesShuffle
|
||||||
|
{
|
||||||
|
template <typename TYPE>
|
||||||
|
void operator()(TYPE *array)
|
||||||
|
{
|
||||||
|
CODE::Xorshift32 prng;
|
||||||
|
for (int i = 0; i < SIZE-1; ++i)
|
||||||
|
std::swap(array[i], array[i + prng() % (SIZE - i)]);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <int SIZE>
|
||||||
|
class ReverseFisherYatesShuffle
|
||||||
{
|
{
|
||||||
int seq[SIZE-1];
|
int seq[SIZE-1];
|
||||||
public:
|
public:
|
||||||
ReversibleFisherYatesShuffle()
|
ReverseFisherYatesShuffle()
|
||||||
{
|
{
|
||||||
CODE::Xorshift32 prng;
|
CODE::Xorshift32 prng;
|
||||||
for (int i = 0; i < SIZE-1; ++i)
|
for (int i = 0; i < SIZE-1; ++i)
|
||||||
seq[i] = i + prng() % (SIZE - i);
|
seq[i] = i + prng() % (SIZE - i);
|
||||||
}
|
}
|
||||||
template <typename TYPE>
|
template <typename TYPE>
|
||||||
void forward(TYPE *array)
|
void operator()(TYPE *array)
|
||||||
{
|
|
||||||
for (int i = 0; i < SIZE-1; ++i)
|
|
||||||
std::swap(array[i], array[seq[i]]);
|
|
||||||
}
|
|
||||||
template <typename TYPE>
|
|
||||||
void reverse(TYPE *array)
|
|
||||||
{
|
{
|
||||||
for (int i = SIZE-2; i >= 0; --i)
|
for (int i = SIZE-2; i >= 0; --i)
|
||||||
std::swap(array[i], array[seq[i]]);
|
std::swap(array[i], array[seq[i]]);
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue