要素数が出来るだけ均等になるように、配列をn個に分割する

split_into.rb

# -*- coding: utf-8 -*-

class Array
  # 配列を要素数が出来るだけ均等になるようにn個に分割
  def split_into(n)
    sizes = (0...n).collect { |i|
      (size % n >= (i+1)) ? (size / n + 1) : (size / n)
    }

    (0...n).collect { |i| slice(sizes[0,i].inject(0, :+), sizes[i]) }
  end
end