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

1.问题描述

LeetCode——515. 在每个树行中找最大值_第1张图片

2.解决办法

广度优先遍历:(经常使用)

  • 创建队列,队列是先进先出,并将根节点放入队列。
  • 判断根节点是否为空,为空直接返回res(ArrayList)
  • 还是以此放入队列 定义一个max同时赋值为Integer.MIN_VALUE,只要队列的数比他大就把值赋给max
  • 当一层遍历完,也就找到了第一层最大的数,放入res中
  • 接着再将max赋值为Integer.MIN_VALUE,再次遍历该层,后面以此类推。

3.代码实现

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

你可能感兴趣的:(算法,leetcode,算法,职场和发展)