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

坚持第26天，加油加油！！！

题目

给定一棵二叉树的根节点 root ，请找出该二叉树中每一层的最大值。

代码

/**
 * 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;
 *     }
 * }
 */

class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> sums = new ArrayList<Integer>();
        dfs(root, 0,  sums);
        return sums;
    }
    public void dfs(TreeNode root, int level, List<Integer> sums) {
        if (root == null) {
            return;
        }
        if (level < sums.size()) {
            sums.set(level, Math.max(sums.get(level),root.val));
        } else {
            sums.add(root.val);
        }
        dfs(root.left, level + 1, sums);
        dfs(root.right, level + 1, sums);
    }
}

反思

1、这里使用了深度遍历，其中的sums的个数就是层数，记录每层的一些特征，下标为i对应数值就是第i-1层的相应特征（1层开始算）