В Ruby работа с массивами — обычная задача. Иногда вы можете столкнуться с ситуацией, когда вам нужно сгладить массив, но только до одного уровня. Это означает, что вложенные массивы должны быть сглажены, но любые дальнейшие вложенные массивы внутри них должны оставаться нетронутыми. В этой статье блога мы рассмотрим различные методы достижения этой цели в Ruby, а также приведем примеры кода для каждого метода.

Метод 1: использование Array#flatten

Встроенный метод Array#flattenRuby — удобный способ выравнивания массивов. По умолчанию он выравнивает весь массив, но мы можем ограничить выравнивание одним уровнем, передав необязательный аргумент.

Пример кода:

arr = [1, 2, [3, [4, 5]], 6, [7, 8, [9]]]
result = arr.flatten(1)
puts result.inspect
# Output: [1, 2, 3, [4, 5], 6, 7, 8, [9]]

Метод 2: использование Array#flat_map и Array#concat

Метод Array#flat_mapобъединяет этапы сопоставления и выравнивания массива за один раз. Мы можем использовать его вместе с Array#concatдля достижения одноуровневого эффекта выравнивания.

Пример кода:

arr = [1, 2, [3, [4, 5]], 6, [7, 8, [9]]]
result = arr.flat_map { |element| element.is_a?(Array) ? element : [element] }
puts result.inspect
# Output: [1, 2, 3, [4, 5], 6, 7, 8, [9]]

Метод 3: использование Array#inject

Метод Array#inject, также известный как Array#reduce, позволяет нам накапливать значение путем перебора элементов массива. Мы можем использовать его для выравнивания массива до одного уровня при создании нового массива.

Пример кода:

arr = [1, 2, [3, [4, 5]], 6, [7, 8, [9]]]
result = arr.inject([]) { |acc, element| acc.concat(element.is_a?(Array) ? element : [element]) }
puts result.inspect
# Output: [1, 2, 3, [4, 5], 6, 7, 8, [9]]

Метод 4. Использование рекурсии

Мы также можем свести массив до одного уровня, используя рекурсивный подход. Этот метод включает в себя перебор элементов и рекурсивное выравнивание любых вложенных массивов.

Пример кода:

def flatten(array)
  array.each_with_object([]) do |element, result|
    result.concat(element.is_a?(Array) ? flatten(element) : [element])
  end
end
arr = [1, 2, [3, [4, 5]], 6, [7, 8, [9]]]
result = flatten(arr)
puts result.inspect
# Output: [1, 2, 3, [4, 5], 6, 7, 8, [9]]

В этой статье мы рассмотрели несколько методов выравнивания массива до одного уровня в Ruby. В примерах рассматривались встроенные методы, такие как Array#flatten, а также пользовательские подходы с использованием Array#flat_map, Array#injectи рекурсии. В зависимости от вашего конкретного варианта использования вы можете выбрать метод, который лучше всего соответствует вашим потребностям.

Используя эти методы, вы можете эффективно манипулировать массивами в Ruby и достигать желаемой плоской структуры. Хорошее понимание манипуляций с массивами необходимо любому разработчику Ruby.

Не забудьте оптимизировать код с учетом размера и сложности массивов, чтобы обеспечить эффективное выполнение. Приятного кодирования!