树——递归思想

1.递归

树要么是空树,要么有两个指针,每个指针指向一棵树。树是一种递归结构,很多树的问题可以使用递归来处理。
在这里插入图片描述

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if root == None:
            return 0
        leftDepth = self.maxDepth(root.left) + 1
        rightDepth = self.maxDepth(root.right) + 1
        return max(leftDepth, rightDepth)

在这里插入图片描述

# 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: TreeNode) -> bool:
        if root == None:
            return True
        if abs(self.height(root.left)-self.height(root.right))<=1:
            return self.isBalanced(root.left) and self.isBalanced(root.right)
        else:
            return False
            

    def height(self, root):
        if root == None:
            return 0

        left_height = self.height(root.left) + 1
        right_right = self.height(root.right) + 1
        return max(left_height, right_right)

在这里插入图片描述

# 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 diameterOfBinaryTree(self, root: TreeNode) -> int:
        self.ans = 1
        self.height(root)
        return self.ans - 1

    def height(self,root):
        if root is None:
            return 0
        left = self.height(root.left)
        right = self.height(root.right)
        self.ans = max(self.ans, left + right + 1)
        return max(left, right) + 1

函数里面的self.ans相当于类里面的全局变量,其他函数也可以引用
记得存下ans,因为最长的可能在某一个子树里面,而不是根节点直接左子树加上右子树
在这里插入图片描述

class Solution:
    def invertTree(self, root: TreeNode) -> TreeNode:
        if root is None:
            return 
        root.left, root.right = root.right, root.left
        self.invertTree(root.left)
        self.invertTree(root.right)
        return root 

在这里插入图片描述

# 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 mergeTrees(self, root1: TreeNode, root2: TreeNode) -> TreeNode:
        root = TreeNode()
        if root1 is None and root2 is None:
            return
        if root1 is None:
            return root2
        elif root2 is None:
            return root1
        else:
            root.val = root1.val + root2.val
            root.left = self.mergeTrees(root1.left, root2.left)
            root.right = self.mergeTrees(root1.right, root2.right)
        return root

在这里插入图片描述

# 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 hasPathSum(self, root: TreeNode, targetSum: int) -> bool:
        if root is None:
            return False
        if root.left is None and root.right is None and root.val == targetSum:
            return True
        return self.hasPathSum(root.left, targetSum-root.val) or self.hasPathSum(root.right, targetSum-root.val) 

在这里插入图片描述
路径不一定以 root 开头,也不一定以 leaf 结尾,但是必须连续。

# 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 pathSum(self, root: TreeNode, sum: int) -> int:
        if root is None:
            return 0
        return self.helper(root,sum)+self.pathSum(root.left,sum)+self.pathSum(root.right,sum)

    def helper(self, root, sum):
        if root is None:
            return 0
        res = 0
        if root.val == sum:
            res = res + 1
        res = res + self.helper(root.left,sum-root.val) + self.helper(root.right,sum-root.val)
        return res

在这里插入图片描述
在这里插入图片描述

class Solution:
    def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
        def helper(root1, root2):
            if root1 is None and root2 is None:
                return True
            if root1 is None or root2 is None:
                return False

            if root1.val != root2.val:
                return False
            else:
                return helper(root1.left, root2.left) and helper(root1.right, root2.right)
        

        if s is None or t is None:
            return False
        
        return helper(s,t) or self.isSubtree(s.left,t) or self.isSubtree(s.right,t) 

在这里插入图片描述

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:

        def sys(root1, root2):
            if root1 is None and root2 is None:
                return True
            if root1 is None or root2 is None:
                return False
            if root1.val != root2.val:
                return False
            return sys(root1.left, root2.right) and sys(root1.right, root2.left)
        
        if root is None:
            return True
        return sys(root.left, root.right)

在这里插入图片描述

class Solution:
    def minDepth(self, root: TreeNode) -> int:
        if root is None:
            return 0

        if root.left is None and root.right is None:
            return 1
        
        if root.left is None:
            return self.minDepth(root.right) + 1
        
        elif root.right is None:
            return self.minDepth(root.left) + 1

        else:
             return min(self.minDepth(root.right), self.minDepth(root.left)) + 1

在这里插入图片描述

class Solution:
    def sumOfLeftLeaves(self, root: TreeNode) -> int:
        if root is None:
            return 0

        left = root.left
        right = root.right

        if left and left.left is None and left.right is None:
            return left.val + self.sumOfLeftLeaves(left) + self.sumOfLeftLeaves(right)
        else:
            return self.sumOfLeftLeaves(left) + self.sumOfLeftLeaves(right)

在这里插入图片描述

class Solution:
    def longestUnivaluePath(self, root: TreeNode) -> int:
        self.res = 0
        self.helper(root)
        return self.res

    def helper(self, root):
        
        if root is None:
            return 0
        
        left = self.helper(root.left)
        right = self.helper(root.right)
        left_tmp = 0
        right_tmp = 0
        if root.left and root.left.val == root.val:
            left_tmp = left + 1
        if root.right and root.right.val == root.val:
            right_tmp = right + 1

        self.res = max(self.res, left_tmp + right_tmp)
        return max(left_tmp, right_tmp)

在这里插入图片描述

class Solution:
    def findSecondMinimumValue(self, root: TreeNode) -> int:
        # 有子节点就是最小的
        if root is None:
            return -1
        if root.left is None and root.right is None:
            return -1

        leftval = root.left.val
        rightval = root.right.val

        if root.left.val == root.val:
            leftval = self.findSecondMinimumValue(root.left)
        if root.right.val == root.val:
            rightval = self.findSecondMinimumValue(root.right)
        if leftval != -1 and rightval != -1:
            return min(leftval, rightval)
        if leftval == -1:
            return rightval
        return leftval
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值