剑指 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;
};
时间和空间复杂度如图