2023.6.16
首先考虑找到待删除节点后应当怎么办,其实就是两种情况:
1.待删除节点的右节点不为None,则将其左节点接到右节点的最左侧,然后将右节点连到删除节点的父节点即可。
2.待删除节点的右节点不存在为None,则直接将左节点连到删除节点的父节点即可。
本来我的考虑是如果要删除当前节点,应当要拿到当前节点的父节点才行,但实际上只能说递归真的很神奇。
首先是递归的返还条件是遇到空节点
之后是递归的逻辑,如果当前节点是待删除节点,返还删除当前节点后的新节点,如果当前节点不是待删除节点,遍历当前节点的左节点或右节点,同时返还值是当前节点。
class Solution:
def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
if not root:
return root
# 中
if root.val == key:
if root.right == None: # 右面是空的
return root.left
elif root.left == None:
return root.right
else:
keeper = right = root.right
while right.left:
right = right.left
right.left = root.left
return keeper
if root.val > key:
root.left = self.deleteNode(root.left, key)
else: root.right = self.deleteNode(root.right, key)
return root