lc_bfs_515_largestValues

最新推荐文章于 2024-07-30 19:30:52 发布

卖山竹的山兔

最新推荐文章于 2024-07-30 19:30:52 发布

阅读量91

点赞数

分类专栏： LeetCode bfs

本文链接：https://blog.csdn.net/weixin_44683593/article/details/107149731

版权

LeetCode 同时被 2 个专栏收录

81 篇文章 0 订阅

订阅专栏

bfs

19 篇文章 0 订阅

订阅专栏

题目：在每个树行中找最大值 middle

您需要在二叉树的每一行中找到最大的值。

示例：
输入:

          1
         / \
        3   2
       / \   \
      5   3   9

输出: [1, 3, 9]

思路：
经典bfs，只需再每层遍历时比较找出每层的最大值

package leetCode.BFS;

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

public class lc_bfs_515_largestValues {

    public List<Integer> largestValues(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        if (root == null)
            return ans;
        ArrayDeque<TreeNode> q = new ArrayDeque<>();
        q.offer(root);
        int max = 0;
        while (!q.isEmpty()) {
            int size = q.size();
            max = q.peek().val;//遍历每层时将最大值初始化为第一个
            for (int i = 0; i < size; i++) {
                TreeNode cur = q.poll();
                if (cur.val > max)
                    max = cur.val;
                if (cur.left != null)
                    q.offer(cur.left);
                if (cur.right != null)
                    q.offer(cur.right);
            }
            ans.add(max);
        }
        return ans;

    }


    class TreeNode {

        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }

    }
}