LeetCode 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
思路
首先要注意是判断左叶子,而不是二叉树左侧节点,所以层序遍历不考虑。
左叶子节点:某节点的左孩子不为空,且左孩子的左右孩子都为空(说明是叶子节点),那么该节点的左孩子为左叶子节点。
由这个定义我们可以知道,判断当前节点是不是左叶子是无法判断的,必须要通过节点的父节点来判断其左孩子是不是左叶子(这一点与之前的题目不同)。所以判断的逻辑为:如果该节点的左节点不为空,该节点的左节点的左节点为空,该节点的左节点的右节点为空,则找到了一个左叶子。
递归三部曲:
- 确定递归函数的参数和返回值。判断一个树的左叶子节点之和,那么一定要传入树的根节点,递归函数的返回值为数值之和。
- 确定终止条件。如果遍历到空节点,那么左叶子值一定是0。如果当前遍历的节点是叶子节点,那其左叶子也必定是0。
- 确定单层递归的逻辑。当遇到左叶子节点的时候,记录数值,然后通过递归求取左子树左叶子之和,和右子树左叶子之和,相加便是整个树的左叶子之和。对于叶子节点来说,其左子树和右子树的和为0。
代码
C++版:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
// 递归法,后序遍历
int sumOfLeftLeaves(TreeNode* root) {
// 终止条件
if(root == NULL) return 0;
// 遇到叶子结点时就可以终止了
if(root->left==NULL && root->right==NULL) return 0;
int leftValue=sumOfLeftLeaves(root->left); // 左
int rightValue=sumOfLeftLeaves(root->right); // 右
// 判断是否有左叶子节点
if(root->left!=NULL && root->left->left==NULL && root->left->right==NULL){
leftValue = root->left->val;
}
int sum=leftValue+rightValue; // 中
return sum;
}
};
Python版:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
# 递归法,后序遍历
def sumOfLeftLeaves(self, root: Optional[TreeNode]) -> int:
if root is None:
return 0
if root.left is None and root.right is None:
return 0
leftValue = self.sumOfLeftLeaves(root.left) # 左
if root.left and not root.left.left and not root.left.right: # 左子树是左叶子的情况
leftValue = root.left.val
rightValue = self.sumOfLeftLeaves(root.right) # 右
sumValue = leftValue + rightValue # 中
return sumValue
需要注意的地方
1.过去解二叉树的题目时,常常是通过节点的左右孩子判断来本节点的属性,而本题我们要通过节点的父节点判断本节点的属性。