力扣:404. 左叶子之和

题目链接:404. 左叶子之和
题目

给定二叉树的根节点 root ,返回所有左叶子之和。

示例 1:
力扣:404. 左叶子之和_第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;
    }
};

你可能感兴趣的:(#,二叉树,leetcode,深度优先,算法)