added reversible Fisher-Yates shuffle

This commit is contained in:
Ahmet Inan 2023-03-25 09:56:27 +01:00
commit e0857552bb

39
permute.hh Normal file
View file

@ -0,0 +1,39 @@
/*
Reversible permutations
Copyright 2023 Ahmet Inan <inan@aicodix.de>
*/
#pragma once
#include "xorshift.hh"
namespace CODE {
template <int SIZE>
class ReversibleFisherYatesShuffle
{
int seq[SIZE-1];
public:
ReversibleFisherYatesShuffle()
{
CODE::Xorshift32 prng;
for (int i = 0; i < SIZE-1; ++i)
seq[i] = i + prng() % (SIZE - i);
}
template <typename TYPE>
void forward(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)
std::swap(array[i], array[seq[i]]);
}
};
}