Ruby技术问询：如何按指定非固定长度拆分数组为子数组

当然有啦！Ruby的设计就是让这类数组操作变得简洁又优雅，给你分享几个非常Rubyic的实现方式👇

方法一：简洁直观的原地修改法 #

如果不介意修改原数组x，一行代码就能搞定：

x = (1..10).to_a
y = [1,4,2,3]

result = y.map { |n| x.slice!(0, n) }
# 输出：[[1], [2,3,4,5], [6,7], [8,9,10]]

原理说明 #

slice!(0, n)是Ruby数组的原生方法，它会从数组开头移除n个元素并返回这部分元素组成的子数组。配合map遍历长度数组y，刚好能按需求依次拆分原数组，代码简洁到极致，完全符合Ruby的风格。

如果需要保留原数组x不被修改，只需要先复制一份副本操作即可：

x = (1..10).to_a
y = [1,4,2,3]

temp_x = x.dup
result = y.map { |n| temp_x.slice!(0, n) }
# 此时原数组x依然保持不变

方法二：无副作用的索引追踪法 #

如果不想对任何数组做原地修改，可以用inject来追踪当前拆分的起始索引，实现纯函数式的操作：

x = (1..10).to_a
y = [1,4,2,3]

result = y.inject([0, []]) do |(current_idx, result_arr), slice_len|
  result_arr << x[current_idx, slice_len]
  [current_idx + slice_len, result_arr]
end.last
# 输出：[[1], [2,3,4,5], [6,7], [8,9,10]]

原理说明 #

• inject的初始值是[0, []]，其中第一个元素是当前拆分的起始索引，第二个是用来存储结果的空数组。
• 每次迭代时，我们从current_idx位置取长度为slice_len的子数组加入结果，然后更新索引为current_idx + slice_len，继续下一次拆分。
• 最后通过.last取出结果数组即可，整个过程不会修改原数组x，非常安全。

两种方法都很Rubyic，你可以根据是否需要保留原数组的需求来选择～

内容的提问来源于stack exchange，提问作者cdDC