这是树的第29篇算法,力扣链接。
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
- 首先找到需要删除的节点;
- 如果找到了,删除它。
示例 1:
输入:root = [5,3,6,2,4,null,7], key = 3 输出:[5,4,6,2,null,null,7] 解释:给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。 一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。 另一个正确答案是 [5,2,6,null,4,null,7]。
递归最适合接这种题型:
func deleteNode(root *TreeNode, key int) *TreeNode {
switch {
case root == nil:
return nil
case root.Val > key:
root.Left = deleteNode(root.Left, key)
case root.Val < key:
root.Right = deleteNode(root.Right, key)
case root.Left == nil || root.Right == nil:
if root.Left != nil {
return root.Left
}
return root.Right
default:
successor := root.Right
for successor.Left != nil {
successor = successor.Left
}
successor.Right = deleteNode(root.Right, successor.Val)
successor.Left = root.Left
return successor
}
return root
}
迭代做法如下:
func deleteNode(root *TreeNode, key int) *TreeNode {
var cur, curParent *TreeNode = root, nil
for cur != nil && cur.Val != key {
curParent = cur
if cur.Val > key {
cur = cur.Left
} else {
cur = cur.Right
}
}
if cur == nil {
return root
}
if cur.Left == nil && cur.Right == nil {
cur = nil
} else if cur.Right == nil {
cur = cur.Left
} else if cur.Left == nil {
cur = cur.Right
} else {
successor, successorParent := cur.Right, cur
for successor.Left != nil {
successorParent = successor
successor = successor.Left
}
if successorParent.Val == cur.Val {
successorParent.Right = successor.Right
} else {
successorParent.Left = successor.Right
}
successor.Right = cur.Right
successor.Left = cur.Left
cur = successor
}
if curParent == nil {
return cur
}
if curParent.Left != nil && curParent.Left.Val == key {
curParent.Left = cur
} else {
curParent.Right = cur
}
return root
}