Leetcode DAY 22:二叉搜索树的最近公共祖先 and 二叉搜索树中的插入操作 and 删除二叉搜索树中的节点

Petal990912

于 2022-12-12 14:44:17 发布

阅读量63

点赞数

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_44189622/article/details/128226078

版权

235. 二叉搜索树的最近公共祖先

由于是二叉搜索树如果root的val处于p和q的val之间就直接返回root

如果root.val 比 p.val和q.val都大那么返回对root.left的递归

如果root.val 比 p.val和q.val都小那么返回对root.right的递归

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if root.val > p.val and root.val > q.val:
            return self.lowestCommonAncestor(root.left, p, q)
        if root.val < p.val and root.val < q.val:
            return self.lowestCommonAncestor(root.right, p, q)
        return root

701.二叉搜索树中的插入操作

class Solution:
    def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        if not root:
            node = TreeNode(val)
            return node
        if root.val < val:
            root.right = self.insertIntoBST(root.right, val)
        if root.val > val:
            root.left = self.insertIntoBST(root.left, val)
        return root