通关剑指 Offer——剑指 Offer II 044. 二叉树每层的最大值

最新推荐文章于 2024-06-28 09:48:33 发布

SK_Jaco

最新推荐文章于 2024-06-28 09:48:33 发布

阅读量183

点赞数 1

分类专栏： # 通关剑指Offer 文章标签： leetcode 算法 java 数据结构

本文链接：https://blog.csdn.net/qq_38550836/article/details/127229723

版权

通关剑指Offer 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

1.题目描述

剑指 Offer II 044. 二叉树每层的最大值

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

示例1：

输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
解释:
          1
         / \
        3   2
       / \   \  
      5   3   9

示例2：

输入: root = [1,2,3]
输出: [1,3]
解释:
          1
         / \
        2   3

示例3：

输入: root = [1]
输出: [1]

示例4：

输入: root = [1,null,2]
输出: [1,2]
解释:      
           1 
            \
             2

示例5：

输入: root = []
输出: []

2.解题思路与代码

2.1 解题思路

要求每一层的最大值，自然想到采用层序遍历会更简单一点。使用队列来完成层序遍历，首先将跟节点放入队列中，并且使用变量 max 记录当前层最大的数，并用 time 记录往队列中存放的次数。由层序遍历的特性可以知道当放入队列次数等于队列中节点个数时，此时该层遍历完成，即该层所有节点都在队列中。因此当 time 等于队列大小时，将 max 放入结果列表，并重置 max 和 time。然后从队列弹出队头，再将弹出节点的左右子节点放入队列中，放入的同时 time 加一并计算最大值 max。最后层序遍历完成之后返回结果数组即可。

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 time = 1;
        int max = root.val;
        while (!queue.isEmpty()) {
            if (queue.size() == time) {
                time = 0;
                ans.add(max);
                max = Integer.MIN_VALUE;
            }
            TreeNode poll = queue.poll();
            if (poll.left != null) {
                queue.offer(poll.left);
                time++;
                max = Math.max(max, poll.left.val);
            }
            if (poll.right != null) {
                queue.offer(poll.right);
                time++;
                max = Math.max(max, poll.right.val);
            }
        }
        return ans;

    }
}