如何用种子确定性地改组数组?

时间:2018-05-01 15:27:05

标签: random rust shuffle random-seed

我发现很难确定性地使用Rust中的随机种子对数组进行洗牌。我想要实现的目标(伪代码):

let v = vec![0, 1, 2, 3];
pseudo_shuffle(v, randomSeed1) // always produces e.g. [3,1,2,0]
pseudo_shuffle(v, randomSeed2) // always produces e.g. [0,2,3,1]

In another Stack Overflow answer我学会了如何使用rand::Rng::shuffle()非确定性地对一个向量进行混洗,但它似乎没有提供一个API来将随机种子应用于生成函数,我就是我自己很难想出一个解决方案,但没有采用一些荒谬的n!复杂度算法。

1 个答案:

答案 0 :(得分:5)

使用实现特征SeedableRng的随机数生成器,并使用所需种子调用from_seed

示例:

use rand::{seq::SliceRandom, SeedableRng}; // 0.6.5
use rand_chacha::ChaChaRng; // 0.1.1

fn main() {
    let seed = [0; 32];
    let mut rng = ChaChaRng::from_seed(seed);

    let mut v1 = vec![1, 2, 3, 4, 5];
    v1.shuffle(&mut rng);
    assert_eq!(v1, [3, 5, 2, 4, 1]);
}

在使用之前克隆RNG或使用相同的种子从头开始创建新的RNG以重置回原始状态。

您也可能对ReseedingRng感兴趣。