Ruby 仅展平 1 级

在 Ruby 中，我们经常需要展平一个数组，从而得到一个不包含嵌套数组的单层数组。然而，有时候我们只需要展平数组的第一层，保留嵌套数组的结构。这个过程叫做“仅展平 1 级”。

实现方法

Ruby 中有多种方式来实现仅展平 1 级的数组。下面是其中两种方法的示例代码：

1. 使用 Array#flatten 方法

[1, [2, [3]]].flatten(1) #=> [1, 2, [3]]

Array#flatten 方法可以将嵌套数组展平成一个单层数组。其中，括号里面的参数表示展平的层数。如果将参数设置为 1，那么只会展平第一层。

2. 使用 Array#flat_map 方法

[1, [2, [3]]].flat_map { |x| x } #=> [1, 2, [3]]

Array#flat_map 方法可以将一个数组映射成一个单层数组。其中，代码块里面的 x 表示数组中的每个元素。如果 x 是一个数组，那么 flat_map 方法会将其展平成一个单层数组。

示例

下面是一个展示仅展平 1 级数组的示例程序：

array = [1, [2, 3], [4, [5, 6]], 7]

puts "原数组：#{array}"
puts "仅展平 1 级后的数组：#{array.flat_map { |x| x }}"

运行这个程序，输出结果如下：

原数组：[1, [2, 3], [4, [5, 6]], 7]
仅展平 1 级后的数组：[1, 2, 3, 4, [5, 6], 7]

总结

通过使用 Array#flatten 或 Array#flat_map 方法，我们可以方便地实现仅展平 1 级数组的功能。无论是展平嵌套数组，还是仅展平 1 级，Ruby 都提供了很多便捷的方法帮助我们处理数组。