view raw
Larsenal Larsenal - 7 months ago 27
Ruby Question

How do I summarize array of integers as an array of ranges?

I'd like to take input such as:


and turn it into something like the following:


Each sub-array represents a range of integers.


Functional approach using Enumerable#chunk:

xs.enum_for(:chunk).with_index { |x, idx| x - idx }.map do |diff, group|
  [group.first, group.last]
# => [[1, 2], [4, 7], [9, 9], [13, 13]] 

How it works: once indexed, consecutive elements in the array have the same x - idx, so we use that value to chunk (grouping of consecutive items) the input array. Finally we just need to take the first and last elements of each group to build the pairs.