题目:给定二叉树的根节点 root,返回所有左叶子之和。
该题主要在于判断该节点是否为父节点的左子节点同时也是叶子节点,所以在遍历节点的时候,就直接判断该节点的左子节点是否为叶子节点。遍历采用后序遍历,顺序为 左,右,中。采用后序遍历可以先判断左子节点是否为叶子节点,如果不是进行递归到下一层再判断,然后再遍历右子节点,判断右子节点的左子节点是否为叶子节点。最后,将判断为左叶子节点的值加在 sum 里面返回。
后序遍历递归法:
时间复杂度:O(n)
思路:
- 首先检查当前节点的左子节点是否存在。如果左子节点存在,再检查左子节点是否为叶子节点。
- 如果是叶子节点,将左子节点的值累加到 sum 中。
- 如果不是叶子节点,继续递归遍历左子树。直到到达叶子节点,不一定是左叶子节点。
- 检查当前节点的右子节点存在,如果存在,递归遍历右子树的左右节点。
- 将已经遍历过的左叶子节点值之和 sum 返回。
代码:
/**
* 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 getSum(TreeNode* cur, int &sum) {
if (cur->left) {
// 如果左子节点是叶子节点,那么sum加上左子节点值
if (cur->left->left == nullptr && cur->left->right == nullptr) {
sum += cur->left->val;
} else
// 如果左子节点不是叶子节点,继续向下递归遍历
getSum(cur->left, sum);
}
if (cur->right) {
// 对于右子节点,递归遍历它的左右子节点
getSum(cur->right, sum);
}
return sum;
}
int sumOfLeftLeaves(TreeNode* root) {
int ret = 0;
if (root == nullptr)
return ret;
return getSum(root, ret);
}
};