存在给定的有限长度N的浮点数(在0和1之间),其表示整数0..N-1上的分布函数。我们试图从这个分布中抽取一个随机数。一种方法是在[0,1](浮点)中绘制一个均匀的随机变量,然后计算该数字的逆累积分布函数。
如果分布在数组中,代码看起来像这样:
let matched distribution draw =
let rec matchRest distribution draw start =
if start = Array.length distribution then start-1
else
let left = draw - distribution.[start]
if left <= 0 then start
else matchRest distribution left (start+1)
matchRest distribution draw 0
其中distribution
是分布函数,draw
是统一的[0,1]数。
当分发是任何序列时,如何重写此功能?显然我可以创建一个临时数组,但它似乎不是一个优雅的解决方案...
答案 0 :(得分:5)
您的matched
功能是一个搜索过程。您不必分解序列来搜索适当的索引;来自Seq module的高阶函数可以帮助您:
/// Taken from http://missingfaktor.blogspot.in/2012/02/f-option-cheat-sheet.html
let getOrElse d = function
| Some a -> a
| None -> d
let matched distribution draw =
distribution
|> Seq.scan (fun (s, d0) d -> (s+1, d0-d)) (0, draw)
|> Seq.tryPick (fun (s, d) -> if d <= 0 then Some s else None)
|> getOrElse (Seq.length distribution-1)
在最坏的情况下,您仍然可以通过Seq.length
枚举整个序列。我认为如果可能的话,最好将Seq.length distribution-1
更改为已知常量。