Одношагового уплощение массива в рубине

Question

Я работаю из this exercise в codewars, который должен выравниваться массивом (до одного уровня глубокого), так что, например, я хочу, чтобы получить выходы:

[1,2,3] >> [1,2,3] 

[[1,2],3] >> [1,2,3] 

[[1,[2]],3] >> [1,[2],3] 

я решил использовать «впрыснуть» - где на следующий элемент будет добавлен к нарастающему итогу, если этот элемент является массивом, или просто толкнул, если это не так:

def flatten(array) 
    array.inject([]) {|result,element| element.kind_of?(Array) result.concat(element) : result<<element} 
end 

Может кто-нибудь объяснить, почему я получаю следующая синтаксическая ошибка?

-e:3: syntax error, unexpected tIDENTIFIER, expecting '}' 
... element.kind_of?(Array) result.concat(element) : result< 

Answer 1

Вы получили ошибку потому что вы пропустили в тройном заявлении на ?:

element.kind_of?(Array) ? result.concat(element) : result<<element 

Для вашей цели вы можете использовать Array#flatten с аргументом:

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