【日常训练】515. 在每个树行中找最大值

题目

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

示例1：

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

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

提示：
二叉树的节点个数的范围是 [0,10⁴]
-2³¹ <= Node.val <= 2³¹ - 1

代码

package dayLeetCode;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;

public class dayleetcode515 {
    // bfs 每次将一层的所有点进行扩展
    public List<Integer> largestValues(TreeNode root) {
        if (root == null){
            return new ArrayList<Integer>();
        }
        List<Integer> ansList = new ArrayList<>();
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);

        while (!queue.isEmpty()){
            int len = queue.size();
            int maxVal = Integer.MIN_VALUE;

            // 拿出改层所有的节点，并找到其中的最大值
            while (len > 0){
                len--;
                TreeNode t = queue.poll();
                maxVal = Math.max(t.val, maxVal);
                if (t.left != null){
                    queue.offer(t.left);
                }

                if (t.right != null){
                    queue.offer(t.right);
                }
            }
            ansList.add(maxVal);
        }
        return ansList;
    }
}