目录
235.二叉搜索树的最近公共祖先
相对于 二叉树的最近公共祖先 本题就简单一些了,因为 可以利用二叉搜索树的特性。题目链接/文章讲解:代码随想录
思路
这道题有个很神奇的解法,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.二叉搜索树中的插入操作
本题比想象中的简单,大家可以先自己想一想应该怎么做,然后看视频讲解,就发现 本题为什么比较简单了。题目链接/文章讲解:代码随想录
思路
插入比下一题删除简单多了,都不需要修改树的结构。直接遍历到叶子节点,插入就完事了。
代码
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.删除二叉搜索树中的节点
相对于 插入操作,本题就有难度了,涉及到改树的结构题目链接/文章讲解:代码随想录
思路
事实证明思路很重要,要像高中数学那样学会分类讨论,我看了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