二叉树每层最大值的节点

昨天下午去面试字节的后端，算法题是这一道。

public class GetLevelMax {

    public static void main(String[] args) {
        // 建立一颗二叉树
        TreeNode root = new TreeNode(1);
        root.left  = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.left.right = new TreeNode(5);
        root.right.left = new TreeNode(6);
        root.right.right = new TreeNode(7);
        List<Integer> output = getLevelMax(root);
        for(int i : output)
            System.out.print(i + " ");
    }

    public static List<Integer> getLevelMax(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int num = -1;
        while (!queue.isEmpty()) {
            int level_len = queue.size();
            for (int i=0; i < level_len; i++) {
                TreeNode node = queue.poll();
                num = node.val > num ? node.val : num;
                if (node.left != null) queue.add(node.left);
                if (node.right != null) queue.add(node.right);
            }
            res.add(num);
        }
        return res;
    }

}

方法肯定是修改二叉树的层次遍历，重点就是如何区分队列中的哪些节点是同一层的。所以先记录队列的 size，然后再使用这个 size 控制 for 循环来求出每一层的最大值。