本文已参与「新人创作礼」活动，一起开启掘金创作之路。

1.题目描述

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

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

 

示例1：

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

示例2：

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

 

2.解题思路与代码

2.1 解题思路

题目要求获取二叉树每一层的最大值，那么可以使用二叉树的层序遍历，这里也可以使用到二叉树层序遍历的特性：放入队列的次数等于队列长度时，那么下一层已全部放入队列中。放入左右节点时，计算最大值，并在每一层遍历完成之后讲最大值放入结果列表中，同时重置最大值和放入次数。

2.2 代码

class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        if (root == null){
            return ans;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        int count = 1;
        int max = root.val;
        while (!queue.isEmpty()) {
            if (count == queue.size()) {
                // 遍历完一层之后将这一层最大值放入结果列表，并初始化最大值和放入次数
                ans.add(max);
                max = Integer.MIN_VALUE;
                count = 0;
            }
            TreeNode poll = queue.poll();
            // 放入左右节点时，计算最大值
            if (poll.left != null) {
                max = Math.max(max, poll.left.val);
                queue.offer(poll.left);
                count++;
            }
            if (poll.right != null) {
                max = Math.max(max, poll.right.val);
                queue.offer(poll.right);
                count++;
            }
        }
        return ans;
    }
}

2.3 测试结果

通过测试

3.总结

• 使用二叉树层序遍历
• 每一层遍历完成之后记录这一层的最大值