算法随想第16天

平衡二叉树

  判断一个二叉树是否为平衡二叉树,递归的判断条件是当遍历至某个节点时,首先该节点对应的两个子树需要满足平衡条件(左右子树之间高度不超过1),那么需要获取左右子树的高度,利用递归获取左右子树的高度是否为-1(因为左右子树其中一棵树为-1,那么两者之间的差值一定会超过1,即不满足平衡),然后判断两者之差是否大于1,大于1即返回-1,不大于1则返回两个子树中的最大高度(为了与节点对应的另一节点判断是否平衡)

class Solution(object):
    def isBalanced(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        """
        return self.get_height(root) != -1
    def get_height(self,node): 
        if not node:
            return 0
        leftheight = self.get_height(node.left)  # 左子树的高度(后序遍历)
        rightheight = self.get_height(node.right) # 右子树的高度(后序遍历)
        if leftheight == -1 or rightheight == -1 or abs(leftheight-rightheight) > 1: #左子树的所有子树满足平衡;右子树的所有子树满足平衡;左右子树之间满足平衡;否则返回-1
            return -1
        return 1 + max(leftheight,rightheight) # 该节点满足平衡则返回该节点对应的子树的最大高度,与对应的另一子树判断是否平衡

二叉树的所有路径

  该题用前序遍历记录路径,首先先将节点值append进path中,然后对该节点是否为叶子节点进行判断,遍历至叶子节点(即无左右节点时)将path放入result中表示一条完整路径。如果不是叶子节点则进行左右子节点的迭代,有子节点则进行迭代,需要将path与result传参进去,迭代一次后,需要将path最新的元素pop出来(回溯过程:将路径从子节点返回至节点,以便进行同层另外子节点的路径遍历)。

class Solution(object):
    def binaryTreePaths(self, root):
        """
        :type root: TreeNode
        :rtype: List[str]
        """
        result = []
        path = []
        path, result = self.traver(root,path,result)
        return result

    def traver(self,node,path,result):
        path.append(str(node.val))
        if not (node.left or node.right):
            result.append('->'.join(path))
            return path, result
        if node.left:
            path, result = self.traver(node.left,path,result)
            path.pop()
        if node.right:
            path, result = self.traver(node.right,path,result)
            path.pop()
        return path, result

左叶子之和

解法一(递归)

class Solution(object):
    def sumOfLeftLeaves(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """  
        if not root:
            return 0
        if not (root.left or root.right):  # 当前节点为叶子节点
            return 0 
        if root.left and not (root.left.left or root.left.right): #当前节点为叶子节点的上一节点   #左
            s_left = root.left.val
        else:
            s_left = self.sumOfLeftLeaves(root.left)
        s_right = self.sumOfLeftLeaves(root.right)   # 右
        s = s_left+s_right  #中
        return s

解法二(迭代–前序遍历)

class Solution(object):
    def sumOfLeftLeaves(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        stack = [root]
        s = 0
        while stack:
            cur = stack.pop()
            if cur.left and not (cur.left.left or cur.left.right):
                s += cur.left.val
            if cur.left:
                stack.append(cur.left)
            if cur.right:
                stack.append(cur.right)
        return s
  • 5
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值