404. 左叶子之和
给定二叉树的根节点 root
,返回所有左叶子之和。
示例 1:
输入: root = [3,9,20,null,null,15,7]
输出: 24
解释: 在这个二叉树中,有两个左叶子,分别是 9 和 15,所以返回 24
示例 2:
输入: root = [1]
输出: 0
提示:
- 节点数在 [1, 1000] 范围内
- -1000 <= Node.val <= 1000
思路:DFS
因为我们求的是左叶子节点之和,所以我们要先判断哪个是左叶子节点;
- 题目中的示例2表明只有根节点,虽然也是叶子节点,但不算左叶子节点;
- 所以左叶子节点的条件就是在根节点
root
节点左边,且是叶子节点,这样我们就能判断左叶子节点了 - 只需要取出左叶子节点的
val
值,累加即可; - 如果不是左叶子节点,则向下递归。
代码:(Java、C++)
Java
/*** 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 int sumOfLeftLeaves(TreeNode root) {if(root == null) return 0;int ans = 0;if(root.left != null && root.left.left == null && root.left.right == null){ans = root.left.val;}else{ans = sumOfLeftLeaves(root.left);}return ans + sumOfLeftLeaves(root.right);}
}
C++
/*** 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 int sumOfLeftLeaves(TreeNode root) {if(root == null) return 0;int ans = 0;if(root.left != null && root.left.left == null && root.left.right == null){ans = root.left.val;}else{ans = sumOfLeftLeaves(root.left);}return ans + sumOfLeftLeaves(root.right);}
}
运行结果:
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n),其中
n
是树中的节点个数 - 空间复杂度: O ( n ) O(n) O(n)。空间复杂度与深度优先搜索使用的栈的最大深度相关。在最坏的情况下,树呈现链式结构,深度为
n
,对应的空间复杂度也为 O ( n ) O(n) O(n)。
题目来源:力扣。
放弃一件事很容易,每天能坚持一件事一定很酷,一起每日一题吧!
关注我 leetCode专栏,每日更新!