[LeetCode] 515. Find Largest Value in Each Tree Row

Given the root of a binary tree, return an array of the largest value in each row of the tree (0-indexed).

Example 1:

Input: root = [1,3,2,5,3,null,9]
Output: [1,3,9]

Example 2:

Input: root = [1,2,3]
Output: [1,3]

Example 3:

Input: root = [1]
Output: [1]

Example 4:

Input: root = [1,null,2]
Output: [1,2]

Example 5:

Input: root = []
Output: []

Constraints:

• The number of nodes in the tree will be in the range [0, 104].
• -231 <= Node.val <= 231 - 1

在每个树行中找最大值。

题意是给一个二叉树，请返回每一层里最大的node.val。

这道题其实很简单，BFS和DFS都能做，着重掌握DFS的方法。

BFS

就是普通的层序遍历，遍历的时候记录每一层上的最大值，放入结果集即可。

时间O(n)

空间O(n)

Java实现

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

DFS

DFS的做法是在比较当前结果集的 size 和树的深度 level。如果当前 res.size + 1 == level，说明 DFS 已经在处理下一层了，此时先把下一层的 root 节点放到结果集。在遍历当前层的时候，如果 res.size == level，则需要把结果集中当前层的那个最大值拿出来跟当前遍历到的节点值作比较，并把较大的那一个再更新回结果集。这里有一个有配图的题解，帮助理解 DFS 是怎么运行的。

时间O(n)

空间O(n)

Java实现

class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        // corner case
        if (root == null) {
            return res;
        }
        helper(res, root, 1);
        return res;
    }

    private void helper(List<Integer> res, TreeNode root, int depth) {
        if (root == null) {
            return;
        }
        if (depth == res.size() + 1) {
            res.add(root.val);
        } else {
            res.set(level - 1, Math.max(res.get(level - 1), root.val));
        }
        helper(res, root.left, depth + 1);
        helper(res, root.right, depth + 1);
    }
}

LeetCode 题目总结