在每个树行中找最大值[分层遍历之一的扩展]

前言

树的层次遍历作为遍历基本方式之一，也是很多问题的基础。
分层遍历不同于层次遍历，在于需要主动分层。
分层遍历有两种情况，用size分层；还有种二维数组的，可递归+level来分层。
掌握了分层遍历，对于分层遍历的扩展就很容易解决了，如每行的最大值，用一个遍历记录每行的最大值即可。

一、在每个树行中找最大值

二、分层遍历扩展

package everyday.medium;

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

// 在每个树行中找最大值
public class LargestValues {
    /*
    target：找到每一层的最大值。
    分层遍历 + 每层记录一个最大值，一层遍历完后加入list
     */
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        // 特殊情况，特殊处理。
        if (null == root) return ans;
        // 用队列+size来做到分层遍历，注:不是层次遍历。
        // 习惯用双端队列，毕竟功能强大，但是有却用不着时也是一种资源浪费。
        ArrayDeque<TreeNode> que = new ArrayDeque<>();
        que.add(root);
        int size = que.size();
        int max = 1 << 31;
        while (!que.isEmpty()) {
            TreeNode node = que.poll();
            // 加入下层节点。
            if (null != node.left) que.offer(node.left);
            if (null != node.right) que.offer(node.right);
            // 记录该层最大值。
            max = Math.max(max, node.val);
            // 用size分层
            if (0 == --size) {
                // 加入该层最大值。
                ans.add(max);
                // 更新下一层的最大值。
                max = 1 << 31;
                // 更新下一层的size
                size = que.size();
            }
        }
        return ans;
    }

    // Definition for a binary tree node.
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

}

总结

1）分层遍历有两种情况，一种队列+size分层，一种递归+level分层。

参考文献

[1] LeetCode 在每个树行中找最大值