红宝石 - 阵列弄平

Question

我具有n深度阵列，其中n是大于变量大于或等于2：

[[1,1],[[1,1],[1,1]]] 

我想变平该数组具有完全相同2的深度，这样的：

[[1,1],[1,1],[1,1]] 

任何人都可以想出一个很好的方法来实现这一目标吗？

Answer 1

这应该这样做。

def flatten_after_first(arr) 
    arr.flat_map { |a| a.first.is_a?(Array) ? a.map(&:flatten) : [a] } 
end 

flatten_after_first [[1,1],[[1,1],[1,1]]] 
    #=> [[1, 1], [1, 1], [1, 1]] 

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

Answer 2

试试这个：

def depth_first_flatten array 
    result = [] 
    array.each do |element| 
    if element.first.is_a? Array 
     result += deph(element) 
    else 
     result << element 
    end 
    end 
    result 
end 

# array = [[1,2],[[3,4],[5,6]]] 
# depth_first_flatten(array) 
# 
# OUTPUT: [[1, 2], [3, 4], [5, 6]] 

Answer 3

可能，这将有助于

def flat(array) 
    array.each do |item| 
    if item.is_a?(Array) && item.flatten.count != item.count 
     flat(item) 
    else 
     $arr << item 
    end 
    end 
end 

### 
$arr = [] 
collection = [[1, 1], [[1, 1], [1, 1], [[1, 2], [1, 2, 3]]]] 
flat(collection) 
puts $arr.inspect 

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


$arr = [] 
collection = [[1,1],[[[1,1],[1,1]],[1,1]]] 
flat(collection) 
$arr 
=> [[1, 1], [1, 1], [1, 1], [1, 1]]