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

目录

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

思路

代码

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

思路

代码

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

思路

代码

---

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

相对于 二叉树的最近公共祖先 本题就简单一些了，因为 可以利用二叉搜索树的特性。 

题目链接/文章讲解：代码随想录

视频讲解：二叉搜索树找祖先就有点不一样了！| 235. 二叉搜索树的最近公共祖先_哔哩哔哩_bilibili

思路

        这道题有个很神奇的解法，Carl哥没有讲具体怎么推导，我也举不出反例，欢迎知道为什么的评论区告诉我。

        只要从上到下去遍历，遇到 cur节点是数值在[p, q]区间中则一定可以说明该节点cur就是p 和 q的公共祖先。

        傻眼了，反正我是死记硬背了

代码

from typing import Optional


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

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

---

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

本题比想象中的简单，大家可以先自己想一想应该怎么做，然后看视频讲解，就发现 本题为什么比较简单了。

题目链接/文章讲解：代码随想录

视频讲解：原来这么简单？ | LeetCode：701.二叉搜索树中的插入操作_哔哩哔哩_bilibili

思路

        插入比下一题删除简单多了，都不需要修改树的结构。直接遍历到叶子节点，插入就完事了。        

代码

from typing import Optional


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 root is None:
            node = TreeNode(val)
            return node

        if val < root.val:
            root.left = self.insertIntoBST(root.left, val)
            return root

        elif val > root.val:
            root.right = self.insertIntoBST(root.right, val)
            return root

---

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

相对于 插入操作，本题就有难度了，涉及到改树的结构 

题目链接/文章讲解：代码随想录

视频讲解：调整二叉树的结构最难！| LeetCode：450.删除二叉搜索树中的节点_哔哩哔哩_bilibili

思路

        事实证明思路很重要，要像高中数学那样学会分类讨论，我看了Carl哥的5种分类情况后自己框框就把代码写出来了，一次通过（每日嘻嘻1/1）

       分以下5种情况

• 第一种情况：没找到删除的节点，遍历到空节点直接返回了
• 第二到第五种情况：找到删除的节点
  • 第二种情况：左右孩子都为空（叶子节点），直接删除节点， 返回NULL为根节点
  • 第三种情况：删除节点的左孩子为空，右孩子不为空，删除节点，右孩子补位，返回右孩子为根节点
  • 第四种情况：删除节点的右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点
  • 第五种情况：左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上，返回删除节点右孩子为新的根节点。

代码

from typing import Optional


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:
            if root.left is None and root.right is None:
                return None

            elif root.left is None and root.right is not None:
                return root.right

            elif root.left is not None and root.right is None:
                return root.left

            elif root.left is not None and root.right is not None:
                pos = root.right
                while pos.left:
                    pos = pos.left
                pos.left = root.left
                return root.right

        root.left = self.deleteNode(root.left,key)
        root.right = self.deleteNode(root.right,key)

        return root

        还有一种通用的，看不懂可以打开链接看看。

class Solution:
    def deleteNode(self, root, key):
        if root is None:  # 如果根节点为空，直接返回
            return root
        if root.val == key:  # 找到要删除的节点
            if root.right is None:  # 如果右子树为空，直接返回左子树作为新的根节点
                return root.left
            cur = root.right
            while cur.left:  # 找到右子树中的最左节点
                cur = cur.left
            root.val, cur.val = cur.val, root.val  # 将要删除的节点值与最左节点值交换
        root.left = self.deleteNode(root.left, key)  # 在左子树中递归删除目标节点
        root.right = self.deleteNode(root.right, key)  # 在右子树中递归删除目标节点
        return root