代码随想录算法训练营第22天 | 235. 二叉搜索树的最近公共祖先 + 701.二叉搜索树中的插入操作 + 450.删除二叉搜索树中的节点

最新推荐文章于 2024-09-14 22:25:23 发布

HY.YUE

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量74

点赞数 1

分类专栏： LeetCode 文章标签：算法 leetcode python

本文链接：https://blog.csdn.net/weixin_42286468/article/details/130972915

版权

LeetCode 专栏收录该内容

53 篇文章 5 订阅

订阅专栏

文章介绍了如何在二叉搜索树中进行三种操作：1)找到两个节点的最近公共祖先，利用递归和二叉搜索树的性质；2)插入一个新值到树中，保持二叉搜索树的特性；3)删除一个节点，同时维护树的结构。解决方案包括递归算法和树的遍历。

摘要由CSDN通过智能技术生成

今日任务

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

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

450.删除二叉搜索树中的节点 - Medium

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

题目链接：力扣-235. 二叉搜索树的最近公共祖先

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

提示：递归法；二叉搜索树的性质，找到的第一个值落在[p, q] 之间的节点即为目标节点

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

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.二叉搜索树中的插入操作 - Medium

题目链接：力扣-701. 二叉搜索树中的插入操作

给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。返回插入后二叉搜索树的根节点。输入数据保证，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。你可以返回任意有效的结果。

提示：在二叉搜索树中新增节点，均可以将其加在叶节点处

# 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 insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        if not root: return TreeNode(val)
        parent = None
        def tranversal(root, val):
            nonlocal parent
            if not root: 
                new_node = TreeNode(val)
                if parent.val > val:
                    parent.left = new_node
                elif parent.val < val:
                    parent.right = new_node
                return
            parent = root
            if root.val > val:
                tranversal(root.left, val)
            elif root.val < val:
                tranversal(root.right, val)

        tranversal(root, val)
        return root

450.删除二叉搜索树中的节点 - Medium

题目链接：力扣-450. 删除二叉搜索树中的节点

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点；
如果找到了，删除它。

提示：想明白删除节点的5种情况

# 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 deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
        if not root: return None
        if root.val < key:
            root.right = self.deleteNode(root.right, key)
        elif root.val > key:
            root.left = self.deleteNode(root.left, key)
        else: #找到要删除的节点
            if not root.left: return root.right
            if not root.right: return root.left
            node = root.right
            while node.left: #找到右子树的最小节点
                node = node.left
            node.left = root.left #将左子树挂在右子树最小节点的左侧
            root = root.right #删掉根节点
        return root