本题采用二叉树的层级遍历进行求解,分别处理每一层,从中找到最大值
/**
* 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) {
//如果头节点为空,则返回空列表
if(root==null){
return new ArrayList<>(0);
}
Queue<TreeNode> queue=new LinkedList<>();
//首先将头节点入队
queue.add(root);
//用于存放每一层的最小值的list
List<Integer> list=new ArrayList<>();
//只要队列中有元素,则一直循环
while(!queue.isEmpty()){
//记录当前层的节点个数
int count=queue.size();
int max=Integer.MIN_VALUE;
//对当前层的所有节点进行遍历
for(int i=0;i<count;i++){
//将当前层的节点依次出队,分别单独处理
TreeNode node=queue.poll();
//如果当前节点值大于max,则赋值给max
if(node.val>max){
max=node.val;
}
//如果当前节点的左孩子不为空,则左孩子入队
if(node.left!=null){
queue.add(node.left);
}
//如果当前节点的右孩子不为空,则右孩子入队
if(node.right!=null){
queue.add(node.right);
}
}
//将当前层的最大值添加到list中
list.add(max);
}
// 返回
return list;
}
}