Leetcode1302 层数最深叶子节点的和

最新推荐文章于 2024-09-12 19:02:20 发布

胖丁微笑

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量66

点赞数

分类专栏：算法刷题文章标签：深度优先算法

本文链接：https://blog.csdn.net/wjl0_7/article/details/126392256

版权

算法刷题专栏收录该内容

16 篇文章 0 订阅

订阅专栏

给你一棵二叉树的根节点 root ，请你返回层数最深的叶子节点的和。
在这里插入图片描述

输入：root = [1,2,3,4,5,null,6,7,null,null,null,null,8]
输出：15

两种解法
①深度优先遍历

class Solution {
    Map<Integer, Integer> map = new HashMap<>();
    int max;
    public int deepestLeavesSum(TreeNode root) {
        dfs(root, 0);
        return map.get(max);
    }
    void dfs(TreeNode root, int depth) {
        if (root == null) return ;
        max = Math.max(max, depth);
        map.put(depth, map.getOrDefault(depth, 0) + root.val);
        dfs(root.left, depth + 1);
        dfs(root.right, depth + 1);
    }
}

②广度优先遍历

class Solution {
    public int deepestLeavesSum(TreeNode root) {
        Map<Integer, Integer> map = new HashMap<>();
        Deque<TreeNode> d = new ArrayDeque<>();
        d.addLast(root);
        int depth = 0;
        while (!d.isEmpty()) {
            int sz = d.size();
            while (sz-- > 0) {
                TreeNode node = d.pollFirst();
                map.put(depth, map.getOrDefault(depth, 0) + node.val);
                if (node.left != null) d.addLast(node.left);
                if (node.right != null) d.addLast(node.right);
            }
            depth++;
        }
        return map.get(depth - 1);
    }
}