我有一个由连续的增量(+1)序列组成的数组。这是一个有三个序列的例子:
sequences = [2,3,4,7,12,13,14,15]
我正在尝试获取每个序列的第一个和最后一个元素,并返回包含这些值的数组。从上面的数组中,结果应如下所示:
[[2,4][7,7][12,15]]
我想出了这个非常天真的解决方案,我认为它可以工作,但它只返回第一个序列。知道为什么吗?和/或任何关于整体更好解决方案的建议?
new_array = []
start_point = sequences[0]
end_point = sequences[0]
sequences.map do |element|
if element == end_point + 1
end_point = element
elsif element == end_point
next
else
new_array << [start_point, end_point]
startpoint = element
end_point = element
end
end
return new_array
答案 0 :(得分:5)
您可以使用chunk_while
查找连续的数字:(这也是文档中的示例)
sequences.chunk_while { |i, j| i + 1 == j }.to_a
#=> [[2, 3, 4], [7], [12, 13, 14, 15]]
和map
以及values_at
一起提取每个子数组的第一个和最后一个元素:
sequences.chunk_while { |i, j| i + 1 == j }.map { |a| a.values_at(0, -1) }
#=> [[2, 4], [7, 7], [12, 15]]
或者更详细:
....map { |a| [a.first, a.last] }