将一个数组拆分成x个数组

3

我有一个数组:

arr1 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

我想把arr1分成x个部分,使每个部分尽可能均等且完整。

arr2  = arr1.foo(3)
# => [1, 2, 3, 4][5, 6, 7][8, 9, 10]
each_slice做的是与我想要的相反的事情,它会将数组分成x个元素一组。
arr2 = arr1.each_slice(3)
# => [1, 2, 3][4, 5, 6][7, 8, 9][10]

如果可能的话,我希望不使用像in_groups这样的rails特定方法来实现。

你是指像这样的吗?https://github.com/rails/rails/blob/master/activesupport/lib/active_support/core_ext/array/grouping.rb#L20 - Dmitry Polushkin
尽量简化。 - Spuck
3个回答

6
class Array
  def in_groups(n)
    len, rem = count.divmod(n)
    (0...n).map { | i | (i < rem) ? self[(len+1) * i, len + 1] : self[len * i + rem, len] }
  end
end

2
你可以在这里使用 len, rem = count.divmod(n) - Stefan
需要一些时间来解析所有的内容,但它确实满足了我的需求。 - Spuck

4

另一种方法:

def in_groups(array, n)
  a = array.dup
  n.downto(1).map { |i| a.pop(a.size / i) }.reverse
end

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

in_groups(arr, 1) #=> [[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]]
in_groups(arr, 2) #=> [[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]]
in_groups(arr, 3) #=> [[1, 2, 3, 4], [5, 6, 7], [8, 9, 10]]
in_groups(arr, 4) #=> [[1, 2, 3], [4, 5, 6], [7, 8], [9, 10]]
in_groups(arr, 5) #=> [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]

现在这真是令人惊讶。 - undur_gongor

3
您可以使用递归:

您可以使用递归:

def in_groups(arr, n)
  return [arr] if n == 1
  len = arr.size/n
  [arr[0,len]].concat in_groups(arr[len..-1], n-1)
end

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

in_groups(arr,  1) #=> [[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]] 
in_groups(arr,  2) #=> [[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]] 
in_groups(arr,  3) #=> [[1, 2, 3], [4, 5, 6], [7, 8, 9, 10]] 
in_groups(arr,  4) #=> [[1, 2], [3, 4], [5, 6, 7], [8, 9, 10]] 
in_groups(arr,  5) #=> [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]] 
in_groups(arr,  9) #=> [[1], [2], [3], [4], [5], [6], [7], [8], [9, 10]] 
in_groups(arr, 10) #=> [[1], [2], [3], [4], [5], [6], [7], [8], [9], [10]] 
in_groups(arr, 11) #=> [[], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]] 

编辑1:如果要优先显示最大的群组,请添加.reverse,或者替换倒数第二行代码:

len = (arr.size.to_f/n).ceil

编辑2:以下是@undur答案的一个略微变体,对于那些拥有“B”和“C”类型大脑的人来说可能更容易理解:

class Array
  def in_groups(n)
    size_small, nbr_large = count.divmod(n)
    size_large, nbr_small = size_small+1, n-nbr_large
    nbr_for_large = nbr_large * size_large  
    self[0, nbr_for_large].each_slice(size_large).to_a.concat(
      self[nbr_for_large..-1].each_slice(size_small).to_a)
  end
end

(1..10).to_a.in_groups(3)
  #=> [[1, 2, 3, 4], [5, 6, 7], [8, 9, 10]] 

很好的递归运用。但我认为原帖作者想要从左到右填充切片,即[[1,2,3,4], [5,6,7], [8,9,10]] - Stefan

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接