Scalacheck：生成总数小于某个值的数字列表

Question

我需要使用 ScalaCheck 生成一个给定大小的随机数字列表，其总数小于给定的固定阈值。 换句话说，请执行以下操作：

val threshold = 3000    
val listGenerator = Gen.listOfN(2000, Gen.choose(1,2000)) suchThat { _.sum < theshold }

虽然上面准确地表达了我想要实现的目标，但它不起作用，因为 suchThat 子句最终丢弃了绝大多数生成的值。

Answer 1

好吧，鉴于您的最大值是 3k，并且您想要一个 2000 的列表，最小值为 1，我希望列表主要是 1 和 2。 下面是一种解决方案（使用当前数字只会生成 1 的列表），

val threshold = 3000

val max = 2000
val max = threshold / n

val listGenerator = Gen.listOfN(max, Gen.choose(1,max))

如果您允许 0，数学会有所改变。

Answer 2

假设阈值为T ，列表的长度为N ，您可以通过在0 until T之间选取N + 1不同的排序随机数来生成此列表，然后计算相邻数字之间的差异。

例如，让T = 10和N = 3 ，首先生成 4 个不同的排序数：

1, 4, 6, 9

通过计算相邻数字的差异，你得到：

3, 2, 3

其和为 8，小于 10。

在此之后，您可以拥有以下生成器：

  def gen(T: Int, N: Int): Gen[collection.Seq[Int]] =
    Gen.pick(N + 1, 0 until T).map(_.sortBy(-_)).map { sorted =>
      (sorted.init, sorted.tail).zipped.map(_ - _)
    }