LeetCode: 404.左叶子节点

计算给定二叉树的所有左叶子之和。

示例:

3
/ \
9 20
/ \
15 7

在这个二叉树中,有两个左叶子,分别是 9 和 15,所以返回 24

解析

我们需要找到这样的节点

  • 属于叶子节点
  • 属于父节点的左子节点

方法一:用栈,dfs遍历,用全局变量res作为累积和。遍历的过程中传递该节点是否是左子节点。同时判断左右子节点是否为None,则可以知道是不是左叶子节点。

class Solution:
    def sumOfLeftLeaves(self, root: TreeNode) -> int:
        stack = []
        res = 0
        if not root:
            return res
        stack.append((root, 0))
        while len(stack) != 0:
            p, flag = stack.pop()
            if flag == 1:
                if p.left == p.right == None:
                    res += p.val
            if p.right:
                stack.append((p.right, 0))
            if p.left:
                stack.append((p.left, 1))    
        return res

方法二,用递归遍历。遍历到左叶子节点则加上左叶子节点的值。

class Solution:
    def sumOfLeftLeaves(self, root: TreeNode) -> int:
        self.res = 0
        def walk(p):
            if p:
                if p.left:
                    if p.left.left == p.left.right == None:
                        self.res += p.left.val
                    walk(p.left)
                if p.right:
                    walk(p.right)
        walk(root)
        return self.res

方法三, 仍是递归,没有使用全局变量。而是使用在递归函数内部累积的方式(即有返回值)。遍历到左叶子节点,则返回值就在此基础上加上右节点的遍历。

class Solution:
    def sumOfLeftLeaves(self, root: TreeNode) -> int:
        if root == None:
            return 0
        res = 0
        if root.left:
            if root.left.left == root.left.right == None:
               res += root.left.val
        return res + self.sumOfLeftLeaves(root.left) + self.sumOfLeftLeaves(root.right)

方法四,在递归的过程中,用一个形参记录该节点是否为左孩子点。和用stack遍历类似。

class Solution:
    def sumOfLeftLeaves(self, root: TreeNode) -> int:
        def cal(p, dir):
            if not p:
                return 0
            if p.left == p.right == None:
                if dir == 1:
                    return p.val
                else:
                    pass      
            return cal(p.left, 1) + cal(p.right, 0)
        return cal(root, 0)

方法五,当然还能用bfs遍历,遍历到左叶子节点就加上去。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值