将数组展平为深度之一

Question

我尝试使用一种深度展平方法

  

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

     

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

     

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

     

[1, 2] => [1, 2]

将我的代码更新为

def flatten(array)
    result = [] of typeof(array) | typeof(array.first)

    array.each do |item|
      if item.is_a?(Array)
        result = ([] of typeof(item) | typeof(item.first)) + result
        result.concat(item)
      else
        result << item
      end
    end

    result
  end

但是展平（[1,2]）无法引发异常https://play.crystal-lang.org/#/r/3p6h

看https://github.com/crystal-lang/crystal/issues/5805

Answer 1

为了安全起见，您需要数组类型及其元素类型的并集：

def unwrap(array)
  result = [] of typeof(array)|typeof(array.first)
  array.each do |item|
    if item.is_a?(Array)
      result.concat item
    else
      result << item
    end
  end
  result
end