【算法题】遍历二叉树的所有左叶子节点,求其值之和

题目描述:

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);
    }
};
    原文作者:夏洛的网
    原文地址: https://blog.csdn.net/liuxiao214/article/details/72716485
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞