使用Javascript实现二叉树级顺序遍历

Question

这是一个leetcode问题。

给定一个二叉树，返回其节点值的级序遍历（即从左到右，逐级）。

例如：给定二叉树[3,9,20， null， null，15,7]，

    3
   / \
  9  20
    /  \
   15   7

将其级别顺序遍历返回为：

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

但我正在用JavaScript尝试一种新的方式，而不是完全按照他们的解决方案。到目前为止，我能够打印阵列，但

如何在新行中打印不同的级别

以下是我目前的代码：

var levelOrder = function(root) {
let output = [];
let queue = [];
let currentNode = root;
queue.push(currentNode);
let currentLevel = 1;
while(queue.length){
    
    currentNode = queue.shift();
    currentLevel--; //this will ensure we are adding new lines only on next level
    output.push(currentNode);
    
    if(currentNode.left){
        queue.push(currentNode.left);
    }
    if(currentNode.right){
        queue.push(currentNode.right);
    }
    
    if(currentLevel = 0){
        output = output + '/n'; //Insert a new line
        currentLevel = queue.length; //2
    }
}
return output;
};

输入：[3,9,20，空，空，15,7]，

Expected Output:
[
[3],
[9,20],
[15,7]
]

LeetCode问题链接：BinarytreeTraversingBFS

Answer 1

这是它的递归解决方案

    var levelOrder = function(root) {
    let result =[];
    if(!root)return result;
    if(!root.left && !root.right){
        result.push([root.val]);
        return result;
    }
    const pushIntoResult =(node, level) =>{
        if(!node) return;
        if(!result[level]){
            result.push([]);
        }
        result[level].push(node.val);
        pushIntoResult(node.left, level+1);
        pushIntoResult(node.right, level+1);
    }
    pushIntoResult(root, 0);
    return result;
};

Answer 2

匿名用户

基于您的代码库，我修改它以供工作。

• 添加用于增加输出索引的索引。
• 使用严格的等号操作符代替赋值变量。
• 删除输出=输出'/n'，因为输出是一个数组。

var levelOrder = function (root) {
  let output = [];
  let queue = [];
  let currentNode = root;
  queue.push(currentNode);
  let currentLevel = 1;
  let index = 0; // Add an index for increasing the output index

  while (queue.length) {

    currentNode = queue.shift();
    currentLevel--;

    if (!output[index]) { // Set default is an array for each output element in first time
      output[index] = [];
    }

    output[index].push(currentNode.val);

    if (currentNode.left) {
      queue.push(currentNode.left);
    }

    if (currentNode.right) {
      queue.push(currentNode.right);
    }

    if (currentLevel === 0) { // Use strict equality operator to compare 0
      index++; // increase index
      currentLevel = queue.length;
    }
  }

  return output;
};

Answer 3

我想你快到了。不确定什么output=output'/n' 是用于。

这将通过：

var levelOrder = function(root) {
    const levels = []

    if(!root) {
        return levels
    }

    const queue = [root]
    while (queue.length){
       const queueLength = queue.length
       const level = []

       for(let i = 0; i < queueLength; i++){

           const node = queue.shift()

           if(node.left){
               queue.push(node.left)
           }
           if(node.right){
               queue.push(node.right)
           }

           level.push(node.val)
       }
       levels.push(level)
   }
    return levels
}

• 有关更多详细信息，请参见讨论板。这里有许多公认的解决方案，包括各种语言和解释、高效算法以及渐近时间/空间复杂性分析1,2