题目描述:
leetcode 404. Sum of Left Leaves:
Find the sum of all left leaves in a given binary tree.
Example:
3
/ \
9 20
/ \
15 7
There are two left leaves in the binary tree, with values 9 and 15 respectively. Return 24.
——两种方法,一种是自己用非递归中序遍历写的,另一种是参考博客中的递归算法。
方法一:
—–方法一:在非递归中序遍历中,有两个过程,一个过程是一直遍历左节点入栈直到为空,另一个过程是,访问栈顶,出栈,访问右节点,过程如下:
while (!s.empty() || p)
{
while (p!= NULL)
{
s.push(p);
p = p->left;
}
if (!s.empty())
{
p = s.top();
s.pop();
cout<<p->val<<endl;
p = p->right;
}
}
——而我们访问每一个节点的动作就是过程二中发生,即输出了该节点的值。结合这道题目,我们每访问一个节点p的时候,顺便记录下它的父节点,怎么知道它的父节点呢,就是栈中该节点的下一个节点,也就是说,在pop之后,如果栈不空,再取一个栈顶节点q,判断我们中序遍历访问的节点p是不是左叶子就很容易,只要q的左节点是p,且p的左节点和右节点都为空,则为左叶子。如果栈空了,说明当前访问的节点p是其父节点(已经访问过,且已出栈)的右节点,节点p的左子树也已访问完毕,所以可以忽略此时这种情况下的节点p。综合之,代码如下,用时3ms:
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
stack<TreeNode*> s;
TreeNode *p = root;
TreeNode *q;
int sum = 0;
while (!s.empty() || p)
{
while (p!= NULL)
{
s.push(p);
p = p->left;
}
if (!s.empty())
{
p = s.top();
s.pop();
if (!s.empty())
{
q = s.top();
if (q != NULL && q->left == p && p->right == NULL && p->left == NULL)
{
sum = sum + p->val;
}
}
p = p->right;
}
}
return sum;
}
};
方法二:
——方法二,递归方法,代码如下,用时6ms:
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if (root == NULL)
{
return 0;
}
if (root->left && root->left->left == NULL && root->left->right == NULL)
return root->left->val + sumOfLeftLeaves(root->right);
return sumOfLeftLeaves(root->left) + sumOfLeftLeaves(root->right);
}
};