LeetCode515.在每个树行中找最大值——java

题目描述：

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

示例1：

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

输入: root = [1,2,3] 输出: [1,3]

代码如下：

/**
 * 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) {
          //创建队列
            Queue<TreeNode> que=new LinkedList<>();
            //创建集合
            List<Integer> result=new ArrayList<>();
            if(root==null)
            {
                return result;
            }
            //将根节点放入队列
            que.offer(root);
            while(!que.isEmpty())
            {
                //获取队列未变化前的大小
                int size=que.size();
                //创建新的集合获取满足条件的元素
                List<Integer> temp=new ArrayList<>();
                for(int i=0;i<size;i++)
                {
                    TreeNode node=que.poll();
                    temp.add(node.val);
                    if(node.left!=null)
                    {
                        que.offer(node.left);
                    }
                    if(node.right!=null)
                    {
                        que.offer(node.right);
                    }
                }
                int max=temp.get(0);
                for(int i=0;i<temp.size();i++)
                {
                    if(max<temp.get(i))
                    {
                        max=temp.get(i);
                    }
                }
                result.add(max);
            }
            return result;
    }
}

优化版：

class Solution {
    public List<Integer> largestValues(TreeNode root) {
        if(root == null){
            return Collections.emptyList();
        }
        List<Integer> result = new ArrayList();
        Queue<TreeNode> queue = new LinkedList();
        queue.offer(root);
        while(!queue.isEmpty()){
            int max = Integer.MIN_VALUE;
            for(int i = queue.size(); i > 0; i--){
               TreeNode node = queue.poll();
               max = Math.max(max, node.val);
               if(node.left != null) queue.offer(node.left);
               if(node.right != null) queue.offer(node.right);
            }
            result.add(max);
        } 
        return result;
    }
}