0

我想以前有人问过这个答案，所以我搜索了，但找不到任何东西。诚然，有大量的 Ruby 数组问题，所以它可能就在那里，只是被埋没了。

无论如何，我试图减少范围的叉积，返回满足某些条件的叉积的所有元素的总和。构建一个简单的例子，如果我有一个这样的数组：

[0..1,0..1,0..1]

我想迭代这个集合：

[
  [0,0,0],
  [0,0,1],
  [0,1,0],
  [0,1,1],
  [1,0,0],
  [1,0,1],
  [1,1,0],
  [1,1,1]
]

并根据条件“”返回一个总和return 1 if i[0] == 1 and i[2] == 0（这将给出 2）。在我设计的示例中，我可以这样做：

br = 0..1

br.reduce(0){|sumx, x|
  sumx + br.reduce(0){|sumy, y|
    sumy + br.reduce(0){|sumz, z|
      sumz + (x == 1 and z == 0 ? 1 : 0)
    }
  }
}

，但在实际应用中，范围集可能要大得多，嵌套减少这种方式会变得非常难看。有没有更好的办法？

4

1 回答 1

3

有两个正交任务，尽量不要混淆它们，这样代码就保持模块化。

  1. 如何构建 N 个数组的笛卡尔积。

  2. 如何过滤产品和计数。

使用Array#product获取笛卡尔积：

xs = [0..1, 0..1, 0..1].map(&:to_a)
xss = xs[0].product(*xs[1..-1]) # or xs.first.product(*xs.drop(1))
#=> [[0, 0, 0], [0, 0, 1], [0, 1, 0], ..., [1, 1, 0], [1, 1, 1]]

现在进行过滤和计算：

xss.count { |x, y, z| x == 1 && z == 0 }
#=> 2

这应该有点难看，那是因为我们需要 classmethodArray::product而不是 method Array#product。没问题，让我们把它添加到我们的扩展模块中，最后编写：

Array.product(0..1, 0..1, 0..1).count { |x, y, z| x == 1 && z == 0 }
#=> 2
于 2011-02-19T14:55:58.007 回答