每日刷题 day10 --- 从上到下打印二叉树

103 阅读1分钟

leetcode传送门

image.png

思路: 题目描述的就是我们打印每一层的数据,但是偶数层的数据从右向左写入,很容易想到层次遍历二叉树,然后对遍历结果中奇数行进行翻转(因为index从0开始)即可

/**
* @param {TreeNode} root
* @return {number[][]}
*/
var levelOrder = function(root) {
     let res = []
     let queue = []
     if(!root){ return res}
     queue.push(root)
     while(queue.length){
         let len = queue.length
         let temp = []
         for(let i=0;i<len;i++){
         
             let node = queue.shift()
             temp.push(node.val)
             if(node.left){
                 queue.push(node.left)
             }
             if(node.right){
                 queue.push(node.right)
             }
             
         }
         res.push(temp)
     }
     
      for(let i =0;i<res.length;i++){
      if((i+1)%2 === 0){
          res[i] = res[i].reverse()
         }
      }

  return res

};

还有一种解法是就是奇数行正常遍历,偶数行进行从右到左遍历。我们可以设置一个标志位level判断当前行是奇数行还是偶数行,如果是奇数行,那下一行就是偶数行,我们可以将当前节点的子节点从左到右入栈,然后出栈的时候赋给队列就是从右到左。具体代码如下:

function Print(pRoot) {
  if (pRoot == null) {
    return [];
  }

 let queue = [],
      temp = [],
      res = [],
      level = 0,
      isEven = true,
      isBePrinted = 1;

queue.push(pRoot);

while (queue.length) {
    let node = queue.shift();
    //判断奇、偶行,奇数行
    //奇数行直接push(node.val);,偶数行就unshift(node.val)
    if (isEven) {
      temp.push(node.val);
    } else {
      temp.unshift(node.val);
    }
    //行数加一
    if (node.left) {
      queue.push(node.left);
      level++;
    }
    if (node.right) {
      queue.push(node.right);
      level++;
    }
    //当前被操作行数已遍历完
    isBePrinted--;
    //处理行数、奇偶、temp置空
    if (isBePrinted === 0) {
      res.push(temp);
      temp = [];
      isBePrinted = level;
      level = 0;
      isEven = !isEven;
    }
  }
  return res;
}