红宝石 | 数组类 flatten!() 函数

简介

flatten!() 函数是 Ruby 数组类中的一个方法，它可以将一个多维数组转换为一个一维数组，它会直接改变原数组，并返回 nil。

语法

array.flatten!

参数

无参数。

返回值

返回 nil。

使用示例

arr1 = [1, 2, [3, 4]]
arr1.flatten!
puts arr1.inspect   # => [1, 2, 3, 4]

arr2 = [1, 2, [3, [4, 5]]]
arr2.flatten!
puts arr2.inspect   # => [1, 2, 3, 4, 5]

原理说明

flatten!() 函数的具体实现可以参考以下代码：

def flatten!(level = -1)
  return nil if empty?

  return self unless is_a?(Array)

  each do |e|
    if e.is_a?(Array)
      if level != 0
        e.flatten!(level - 1)
        self[self.index(e)..self.index(e)] = e
      end
    end
  end
  self
end

该实现中，通过递归访问数组中的元素，如果元素也是一个数组，则将其展开，直到得到一个一维的数组。

注意事项

需要注意的是，flatten!() 函数会直接改变原数组，如果不希望改变原数组，可以使用 flatten() 函数来生成一个新的数组。

参考资料

• Ruby Document: Array#flatten!