题目链接: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
解题思路:
方案一:递归
递归的遍历顺序为后序遍历(左右中),是因为要通过递归函数的返回值来累加求取左叶子数值之和。
本题的递归三部曲:
(1)确定递归的参数和返回值:判断一个树的左叶子节点之和,那么一定要传入树的根节点,递归函数的返回值为数值之和,所以为int,使用题目中给出的函数就可以了。
(2)确定终止条件:如果当前节点为空,则返回0。
if (!root) return 0;
(3)确定单层递归逻辑:当遇到左叶子节点的时候,记录数值,然后通过递归求取左子树左叶子之和,和 右子树左叶子之和,相加便是整个树的左叶子之和。
int leftVal = sumOfLeftLeaves(root->left); //递归左子树,返回左子树左叶子的值
int rightVal = sumOfLeftLeaves(root->right); //递归右子树,返回右子树左叶子的值
int curLeftVal = 0; //中
//如果当前节点的左节点为叶子节点
if (root->left && !root->left->left && !root->left->right) {
curLeftVal = root->left->val; //左叶子的值
}
return curLeftVal + leftVal + rightVal;
代码(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) {
//确定终止条件
//如果当前节点为空,则返回0
if (!root) return 0;
//确定单层递归逻辑
int leftVal = sumOfLeftLeaves(root->left); //递归左子树,返回左子树左叶子的值
int rightVal = sumOfLeftLeaves(root->right); //递归右子树,返回右子树左叶子的值
int curLeftVal = 0; //中
//如果当前节点的左节点为叶子节点
if (root->left && !root->left->left && !root->left->right) {
curLeftVal = root->left->val; //左叶子的值
}
return curLeftVal + leftVal + rightVal;
}
};
递归精简版:
//递归精简版
//后序遍历:左右中
class Solution {
public:
//确定递归的参数和返回值
//参数:根节点,返回值:左叶子之和
int sumOfLeftLeaves(TreeNode* root) {
//确定终止条件
//如果当前节点为空,则返回0
if (!root) return 0;
int curLeftVal = 0; //中
//如果当前节点的左节点为叶子节点
if (root->left && !root->left->left && !root->left->right) {
curLeftVal = root->left->val; //左叶子的值
}
return curLeftVal + sumOfLeftLeaves(root->left) + sumOfLeftLeaves(root->right);
}
};
方案二:迭代
//迭代
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
stack<TreeNode*> stk;
if (root) stk.push(root);
int sum = 0; //所有左叶子值之和
while (!stk.empty()) {
TreeNode* cur = stk.top(); //取出栈顶节点
stk.pop();
//如果cur->left是左叶子节点
if (cur->left && !cur->left->left && !cur->left->right) {
sum += cur->left->val;
}
if (cur->right) stk.push(cur->right); //右
if (cur->left) stk.push(cur->left); //左
}
return sum;
}
};