计算给定二叉树的所有左叶子之和。
示例:
来源:力扣(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);
}
};
Comments | 0 条评论