可能重复:
ruby array element grouping
例子。给定数组A:
a = [1, 2, 3]
它的长度是3,所以我想打印所有2长度的数组。这些是:
[1, 2]
[1, 3]
[2, 3]
我不知道ruby中是否有方法来获取子集数组。如果没有这种方法,最有效的方法是什么。
最佳答案
这只是2个元素的简单combination:
>> xs = [1, 2, 3]
>> xs.combination(xs.size - 1).to_a
=> [[1, 2], [1, 3], [2, 3]]
[编辑]正如@joshua在一条评论中指出的,文档声明订单没有保证(!)。所以这里有一个函数实现,它按照您要求的顺序生成组合。为了完整起见,我将使它像原来的
combination
方法一样变懒:require 'enumerable/lazy'
class Array
def combinations_of(n)
if n == 0
[[]].lazy
else
0.upto(self.size - 1).lazy.flat_map do |idx|
self.drop(idx + 1).combinations_of(n - 1).map do |xs|
[self[idx]] + xs
end
end
end
end
end