首页 > 解决方案 > 整数到块数组

问题描述

我需要将整数转换为间隔数组。例如,

number_to_steps(number: 10, step: 3)
# => [[0, 2], [3, 5], [6, 8], [9, 9]]

number_to_steps(number: 7, step: 2)
# => [[0, 1], [2, 3], [4, 5], [6, 6]]

number_to_steps(number: 8, step: 2)
# => [[0, 1], [2, 3], [4, 5], [6, 7]]

我试过:

def number_to_ranges(number:, size:)
    chunks = ((number - 1) / size.to_f).ceil

    (0..chunks - 1).map do |index|
      from = index * size
      to = (index + 1) * size - 1

      [ from, to > number ? number : to ]
    end
  end

但它不能正常工作。例如,

number_to_ranges(number: 14, step: 4)
[[0, 3], [4, 7], [8, 11], [12, 14]]

不应该去14。

任何的想法?

标签: rubyrange

解决方案


另一种方法(您的方法)是使用each_slice.

def number_to_steps(number:, step:)
  (0...number).each_slice(step).map { |arr| [arr.first, arr.last] }
end

测试:

number_to_steps(number: 10, step: 3)
 #=> [[0, 2], [3, 5], [6, 8], [9, 9]]

number_to_steps(number: 7, step: 2)
 #=> [[0, 1], [2, 3], [4, 5], [6, 6]]

number_to_steps(number: 8, step: 2)
 #=> [[0, 1], [2, 3], [4, 5], [6, 7]]

推荐阅读