力扣:404.左叶子之和
代码随想录
如果左节点不为空,且左节点没有左右孩子,那么这个节点就是左叶子。
题目:
给定二叉树的根节点 root ,返回所有左叶子之和。
代码:
先序遍历每一个节点同时判断每一个节点的左孩子是不是左叶子。a用全局变量。
即在遍历节点的同时顺便解题。
class Solution {
public:
int a = 0;
int sumOfLeftLeaves(TreeNode* root) {
if(!root) return a;
if(root->left && !root->left->left && !root->left->right) a += root->left->val;
sumOfLeftLeaves(root->left);
sumOfLeftLeaves(root->right);
return a;
}
};
递归代码②:
所有左叶子之和 = 左树左叶子之和+右树左叶子之和+根左叶子之和。
递归的遍历顺序为后序遍历(左右中),是因为要通过递归函数的返回值来累加求取左叶子数值之和。
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if(!root) return 0;
int midValue = 0;
int leftnum = sumOfLeftLeaves(root->left);
int rightnum = sumOfLeftLeaves(root->right);
if(root->left && !root->left->left && !root->left->right) {
midValue = root->left->val;
}
int sum = midValue + leftnum + rightnum;
return sum;
//return leftnum + rightnum + root->left->val;
}
};
迭代方法:
用栈遍历每一个节点,看每一个节点的左孩子是不是左叶子,是则加上值。
代码:
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;
}
};