Question

问题陈述：

给定二叉树，返回其节点的级别顺序遍历＆＃39; 值。（即，从左到右，逐级）。

For example:
Given binary tree [3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7
return its level order traversal as:
[
  [3],
  [9,20],
  [15,7]
]

我用BFS解决了这个问题，这是最直观的方法。但是，我尝试用另一种方式解决它，而我却无法解决。下面是样本输入/正确输出与我的输出：

Your input
[3,9,20,null,null,15,7]

Your answer
[[3],[[9],[],[20],[[15],[],[7],[]]]]

Expected answer    
[[3],[9,20],[15,7]]

这显然是因为代码[]中的某处正在返回，但这是我的代码：

def level_order(root)
    return [] if root.nil?

    arr = merge([level_order(root.left)], [level_order(root.right)]) #this returns an empty arr if both of those are nil..
    arr.insert(0, [root.val]) 
end

def merge(arr1, arr2)
    i = j = 0
    while i < arr1.length && j < arr2.length
        arr1[i] += arr2[j]
        i += 1
        j += 1
    end
    while j < arr2.length #check if any remaining elements in arr 2
        arr1 << arr2[j]
        j += 1
    end

    arr1
end

在上面，我通过做+ =代替＆lt;＆lt;来处理[]情况如果一个arr为空，则合并功能有效。这里的想法是我合并左侧和右侧的级别顺序遍历的每个级别，然后在数组的开头插入根。

我还认为root可以作为一个空数组插入，但这不可能发生，因为我有一个初始return语句，如果root为nil则调用它。有什么想法吗？

Answer 1

应该像更改此

一样简单

arr = merge([level_order(root.left)], [level_order(root.right)])

要

arr = merge(level_order(root.left), level_order(root.right))

但是我的写法略有不同：

input = [3,9,20,nil,nil,15,7]

output = []
start  = 0
length = 1
while start < input.length do
  output << input.slice(start, length).compact
  start  += length
  length *= 2  
end

puts output.inspect

这样可以避免构建树，并且比递归更有效。

级别顺序遍历二叉树问题

1 个答案: