LeetCode 404. 左叶子之和

博主 872 2020-09-19

计算给定二叉树的所有左叶子之和。

image.png示例:

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-left-leaves
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


这道题没啥好说的,叶子节点就是左右都无孩子的节点,而左叶子节点只需要在dfs的时候,如果向左,则添加一个标记即可

代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    int ans = 0;
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if (root == nullptr) {// 空树
            return 0;
        }else if (root->left == nullptr && root->right == nullptr){ // 只有根节点
            return 0;
        }else{
            dfs(root->left, true);
            dfs(root->right, false);
            return ans;
        }
    }
    
    void dfs(TreeNode* root, bool flag){
        if (root == nullptr) {
            return;
        }
        if (root->left == nullptr && root->right == nullptr && flag) {// 左叶子节点
            ans += root->val;
        }
        dfs(root->left, true);
        dfs(root->right, false);
    }
};