【LeetCode力扣】1302. 层数最深叶子节点的和(Java)

文章目录

  • 一、题目
  • 二、解题
    • 1.思路
    • 2.代码
    • 3.复杂度分析
  • 总结


一、题目

层数最深叶子节点的和中等
给你一棵二叉树的根节点 root ,请你返回 层数最深的叶子节点的和

返回的是 所有层数最深的节点(即所有最深的叶子节点)并计算节点值之和即可。

示例 1:
【LeetCode力扣】1302. 层数最深叶子节点的和(Java)_第1张图片

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

示例 2:

输入:root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出:19

提示:

  • 树中节点数目在范围 [1, 104] 之间。
  • 1 <= Node.val <= 100

二、解题

1.思路

  • 假设根节点为第0层;
  • 如果当前节点层数大于最大层数maxLevel,更新最大层数,并将节点和sum更新为当前节点的值
  • 如果当前节点层数等于最大层数,将当前节点值加到最深节点之和

相同思路也可以解104. 二叉树的最大深度
104解:leetcode104. 二叉树的最大深度(Java)

2.代码

/**
 * 二叉树的定义
 * 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 {
    int sum = 0;
    int maxLevel;
    public int deepestLeavesSum(TreeNode root) {
        dfs(root,0);
        return sum;
    }

    public void dfs(TreeNode root,int level){
        if(root == null){
            return;
        }
        if(maxLevel < level){
            maxLevel = level;
            sum = root.val;
        }else if(maxLevel == level){
            sum += root.val;
        }
        dfs(root.left,level + 1);
        dfs(root.right,level + 1);
    }
}

3.复杂度分析

时间复杂度:O(n),其中 n 是二叉树的节点数。深度优先搜索需要遍历每个节点一次。
空间复杂度:O(n),其中 n 是二叉树的节点数。空间复杂度主要取决于递归调用栈的深度,为二叉树的深度

总结

当前节点处理结束之后,对当前节点的左、右子节点继续深度优先搜索。

你可能感兴趣的:(LeetCode,leetcode,java,深度优先)