二叉树的层级遍历

515. 在每个树行中找最大值

 

本题采用二叉树的层级遍历进行求解，分别处理每一层，从中找到最大值 

/**
 * 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;
    }
}