计算给定二叉树的所有左叶子之和。
思路
是左叶子不是左侧节点。
节点A的左孩子不为空,且左孩子的左右孩子都为空(说明是叶子节点),那么A节点的左孩子为左叶子节点。(这就是判断条件)
根据当前节点判断是否为左叶子不可行,只能通过父节点判断。
if(node->left != NULL && node->left->left == NULL && node->left->right == NULL){
//左叶子节点处理逻辑
}
递归法
1. 递归参数和返回值
传入参数为根节点,返回值为int型数值。使用给出的函数即可。
2. 终止条件
当遍历到空节点停止。
如果当前遍历的是叶子节点,那其左节点也为空,不用继续递归。
3. 单层递归逻辑
遇到左叶子节点时记录数值,然后递归求取左子树左叶子之和和右树左叶子之和。
int leftValue = sumOfLeftLeaves(root->left); // 左
if (root->left && !root->left->left && !root->left->right) {
leftValue = root->left->val;
}
int rightValue = sumOfLeftLeaves(root->right); // 右
int sum = leftValue + rightValue; // 中
return sum;
为什么是左右中呢?因为主要关注的是叶子节点,遍历完左右叶子后求和。
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if(root == nullptr) return 0;
if(root->left == NULL && root->right == NULL) return 0;
int leftVal = sumOfLeftLeaves(root->left);
if(root->left && !root->left->left && !root->left->right){
leftVal = root->left->val;
}
int rightVal = sumOfLeftLeaves(root->right);
int sum = leftVal + rightVal;
return sum;
}
};
迭代法
前中后序遍历都可以,统计做叶子节点。
前序遍历为例:
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
stack<TreeNode*> st;
if (root == NULL) return 0;
st.push(root);
int result = 0;
while (!st.empty()) {
TreeNode* node = st.top();
st.pop();
if (node->left != NULL && node->left->left == NULL && node->left->right == NULL) {
result += node->left->val;
}
if (node->right) st.push(node->right);
if (node->left) st.push(node->left);
}
return result;
}
};
总结
此题的特殊之处在于,需要根据当前节点的父节点来判断当前节点是否为左叶子节点。
根据叶子节点本身是无法判断其自身是否为左叶子节点。
按照题目的要求,自己给出左叶子节点的定义和判断条件是关键。
这道题的迭代法我感觉比递归清楚一点,递归需要继续深入理解。