Haskell RandomGen用于新矩阵

Question

我必须在haskell中实现这个功能：

insertRandomNumber :: RandomGen g => [[Int]] -> g -> ([[Int]], g)

该函数在我的矩阵中的随机位置插入一个随机数。 我只能在有0的位置插入。矩阵大小为4 x 4。 我有这个 ：

insertRandomNumber :: RandomGen g => [[Int]] -> g -> ([[Int]], g)
insertRandomNumber mat g =
    let (pos,_) = randomR (1,16) g
        ok = free (pos `div` 4) (pos `mod` 4) mat
    in if ok == True 
       then newmatrix pos mat 
       else insertRandomNumber mat g

问题是，如果第一个位置不是免费的，我的程序将阻止其他。 我希望你能给我一个如何使用randomgen将随机数插入免费随机位置的例子。

Answer 1

您需要保留新的随机数生成器并传递它而不是旧的生成器。 如果不这样做意味着您将始终在每次迭代中生成相同的pos并永远循环。

在代码中：

let (pos,_) = randomR (1,16) g

在这里，您明确忽略了新的RNG状态。 代替：

let (pos,newGen) = randomR (1,16) g

然后你传递了旧的生成器，它将生成完全相同的pos ：

in if ok == True then newmatrix pos mat else insertRandomNumber mat g

相反，您应该传递新的生成器状态：

in if ok == True then newmatrix pos mat else insertRandomNumber mat newGen

Answer 2

您可以将此问题分为三个部分：枚举包含0的所有索引，从中选择随机索引，并在所选索引处插入随机数。 所以像这样：

insertRandomNumber :: RandomGen g => [[Int]] -> g -> ([[Int]], g)
insertRandomNumber mat g0 = case validPositions mat of 
                             [] -> (mat, g0)
                             xs -> let (i,g1)  = randomR (0, length xs - 1) g0
                                       (v, g2) = randomR (1,100 :: Int) g1 
                                    in (replaceAt mat (xs !! i) v, g2)

我将让你写函数validPositions和replaceAt 。