剑指 Offer II 044. 二叉树每层的最大值 Javascript解法

剑指 Offer II 044. 二叉树每层的最大值

难度：中等

给定一棵二叉树的根节点 root ，请找出该二叉树中每一层的最大值。

示例1：
输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
解释:

          1
         / \
        3   2
       / \   \  
      5   3   9 

示例2：

输入: root = []
输出: []

提示：
二叉树的节点个数的范围是 [0,104]
-2^31 <= Node.val <= 2^31 - 1

思路

层序遍历的思路，只不过不用新建一个临时数组存储每一层的全部变量，而是逐一比较大小，压入最大值即可

层序遍历：用队列先进先出的特性，每遍历一个结点，就将他的左右子结点压入队列(如果有的话)，当每一层遍历完毕，此时队列中结点都为下一层的结点，将这时候的队列长度记为遍历下一层的循环次数即可

代码如下

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var largestValues = function(root) {
    if(!root) return [];
    let queue = [root], result = [], size, max, x;

    while(queue.length) {
        size = queue.length; // 每层结点个数，决定循环次数
        if(size) max = queue[0].val; // 先把每层最大值设为第一个
        for(let i = 0; i < size; i++) {
            x = queue.shift(); // 队头元素移出来
            if(x.val > max) max = x.val; // 比较大小

            if(x.left) queue.push(x.left); // 左右子树继续递归
            if(x.right) queue.push(x.right);
        }
        result.push(max);
    }
    return result;
};

时间和空间复杂度如图