给定一棵二叉树的根节点 root
,请找出该二叉树中每一层的最大值。
示例1:
输入:root = [1,3,2,5,3,null,9]
输出:[1,3,9]
示例2:
输入:root = [1,2,3]
输出:[1,3]
提示:
- 二叉树的节点个数的范围是
[0,104]
231 <= Node.val <= 231 - 1
思路:
- 层次遍历,遍历每一层是,获取最大的值,放入list中。
- 如何知道哪些节点是同一层的?层次遍历的思想是:首先将根节点放入队列中,然后出队→ 左节点入队→右节点入队。定义变量size,size=队列大小,这个size其实就是同一层节点的个数,对size进行while循环,循环出来的就是同一层的节点。
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root == null) {
return res;
}
queue.add(root);
while(! queue.isEmpty()) {
int size = queue.size();
int max = Integer.MIN_VALUE ;
while (size > 0) {
TreeNode poll = queue.poll();
max = Math.max(max, poll.val);
if(poll.left != null) {
queue.add(poll.left);
}
if(poll.right != null) {
queue.add(poll.right);
}
size -- ;
}
res.add(max);
}
return res;
}
}