代码随想录刷题第17天|110.平衡二叉树,257.二叉树的所有路径,404.左叶子之和

文章介绍了如何使用递归和迭代方法在Python中实现对平衡二叉树的判断,包括检查节点是否平衡,找到所有可能的路径,以及计算左叶子节点的和。同时,展示了两种不同的解决方案:递归和迭代。
摘要由CSDN通过智能技术生成

 110.平衡二叉树

当一个节点不满足时,result返回-1

递归法:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isBalanced(self, root: Optional[TreeNode]) -> bool:
        if self.get_height(root)!=-1:
            return True
        else:
            return False
    
    def get_height(self,root):
        if not root:
            return 0#空值返回0,遇到空节点为终止
        left=self.get_height(root.left)
        right=self.get_height(root.right)
        if left==-1 or right==-1 or abs(left-right)>1:
            return -1
        return max(left,right)+1

257.二叉树的所有路径

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def binaryTreePaths(self, root: Optional[TreeNode]) -> List[str]:
        #递归加回溯
        result=[]
        path=[]
        if not root:
            return result
        self.traversal(root,path,result)
        return result

    def traversal(self,cur,path,result):
        path.append(cur.val)#中,最后一个节点也要入path
        if not cur.left and not cur.right:#到达叶子节点
            sPath='->'.join(map(str,path))#转换为字符串
            result.append(sPath)
            return
        if cur.left:#左
            self.traversal(cur.left,path,result)
            path.pop()#回溯
        if cur.right:
            self.traversal(cur.right,path,result)
            path.pop()

        

 迭代法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def binaryTreePaths(self, root: Optional[TreeNode]) -> List[str]:
        stack,path_st,result=[root],[str(root.val)],[]

        while stack:
            cur=stack.pop()
            path=path_st.pop()
            if not(cur.left or cur.right):#不是叶子节点
                result.append(path)
            if cur.right:
                stack.append(cur.right)
                path_st.append(path+'->'+str(cur.right.val))
            if cur.left:
                stack.append(cur.left)
                path_st.append(path+'->'+str(cur.left.val))

        return result  

        

左叶子之和:通过二叉树来判断,平时我们解二叉树的题目时,已经习惯了通过节点的左右孩子判断本节点的属性,而本题我们要通过节点的父节点判断本节点的属性。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def sumOfLeftLeaves(self, root: Optional[TreeNode]) -> int:
        if root is None:
            return 0#遍历到空节点
        if root.left is None and root.right is None:#叶子节点
            return 0
        
        leftValue=self.sumOfLeftLeaves(root.left)
        if root.left and not root.left.left and not root.left.right:
            #左子树是左叶子的情况
            leftValue=root.left.val
        rightValue=self.sumOfLeftLeaves(root.right)#中
        
        sum_val=leftValue+rightValue

        return sum_val

迭代法 

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def sumOfLeftLeaves(self, root: Optional[TreeNode]) -> int:
        if root is None:
            return 0#遍历到空节点
        st=[root]
        result=0
        while st:
            node=st.pop()
            if node.left and node.left.left is None and node.left.right is None:
                result+=node.left.val
            
            if node.left:
                st.append(node.left)
            if node.right:
                st.append(node.right)
        return result


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值