数组到连续数字的范围

时间:2013-09-18 19:01:21

标签: ruby arrays algorithm

我正在尝试创建一个Ruby函数,将一组唯一数字转换为连续数字的范围。

[1, 2, 3, 5, 6, 8, 9] => [(1..3), (5..6), (8..9)]

这似乎并不太难,但我想知道是否有更好的方法。

3 个答案:

答案 0 :(得分:2)

这是你要问的同一个问题的solution。链接的代码比你需要的工作多一些(数字不需要排序或连续),但它会做到这一点。或者,您可以使用此代码,由@NewAlexandria建议:

class Array
   def to_ranges
      compact.sort.uniq.inject([]) do |r,x|
         r.empty? || r.last.last.succ != x ? r << (x..x) : r[0..-2] << (r.last.first..x)
      end
   end
end

答案 1 :(得分:2)

这是如何使用Enumerable#slice_before的?

ar = [1, 2, 3, 5, 6, 8, 9]
prev = ar[0]
p ar.slice_before { |e|
  prev, prev2 = e, prev
  prev2 + 1 != e
}.map{|a| a[0]..a[-1]} 
# >> [1..3, 5..6, 8..9]

ar = [1, 2, 3, 5, 6,7, 8, 9,11]
prev = ar[0]
p ar.slice_before { |e|
  prev, prev2 = e, prev
  prev2 + 1 != e
}.map{|a| a[0]..a[-1]} 
# >> [1..3, 5..9, 11..11]

答案 2 :(得分:1)

这是我在处理IP地址范围时写的一段时间:

class Array
  # [1,2,4,5,6,7,9,13].to_ranges       # => [1..2, 4..7, 9..9, 13..13]
  # [1,2,4,5,6,7,9,13].to_ranges(true) # => [1..2, 4..7, 9, 13]
  def to_ranges(non_ranges_ok=false)
    self.sort.each_with_index.chunk { |x, i| x - i }.map { |diff, pairs|
      if (non_ranges_ok)
        pairs.first[0] == pairs.last[0] ? pairs.first[0] : pairs.first[0] .. pairs.last[0]
      else
        pairs.first[0] .. pairs.last[0]
      end
    }
  end
end

if ($0 == __FILE__)
  require 'awesome_print'

  ary = [1, 2, 4, 5, 6, 7, 9, 13, 12]
  puts ary.join(', ')
  ap ary.to_ranges

  ary = [1, 2, 4, 8, 5, 6, 7, 3, 9, 11, 12, 10]
  puts ary.join(', ')
  ap ary.to_ranges

end

true传递给to_ranges,它不会将单个元素转换为单元素范围。