golang力扣leetcode 450.删除二叉搜索树中的节点

最新推荐文章于 2025-09-10 22:03:14 发布

cheems~

最新推荐文章于 2025-09-10 22:03:14 发布

阅读量317

点赞数

CC 4.0 BY-SA版权

分类专栏： Golang leetcode 文章标签： leetcode golang 数据结构

本文链接：https://blog.csdn.net/qq_42956653/article/details/122666604

Golang leetcode 专栏收录该内容

189 篇文章

订阅专栏

本文介绍了一种从二叉搜索树(BST)中删除指定节点的有效方法。文章详细阐述了三种情况：删除叶子节点、删除有右子节点的节点及删除有左子节点的节点，并提供了具体的实现代码。

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

题解

如果key < root.Val，说明要删除的节点在BST的左子树，那么递归的去左子树删除即可
2 如果key > root.Val，说明要删除的节点在BST的右子树，那么递归的去右子树删除即可
3 如果key = root.Val，说明要删除的节点就是本节点，这一类又分以下三种情况:

要删除的节点是叶子节点，那很简单，直接将当前节点删除，置为nil即可;
要删除的节点有右子节点，那么为了维持BST的特性，我们需要找到该节点的后继节点post(BST中大于它的最小节点)，将该节点的值更新为后继节点post的值，然后递归的在当前节点的右子树中删除该后继节点post;
要删除的节点有左子节点，那么为了维持BST的特性，我们需要找到该节点的前驱节点pre(BST中小于于它的最大节点)，将该节点的值更新为前驱节点pre的值，然后递归的在当前节点的左子树中删除该前驱节点pre;

代码

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

func deleteNode(root *TreeNode, key int) *TreeNode {
	if root == nil {
		return root
	}
	if key < root.Val {
		root.Left = deleteNode(root.Left, key)
	} else if key > root.Val {
		root.Right = deleteNode(root.Right, key)
	} else {
		if root.Left == nil && root.Right == nil {
			root = nil
		} else if root.Right != nil {
			root.Val = successor(root).Val
			root.Right = deleteNode(root.Right, root.Val)
		} else if root.Left != nil {
			root.Val = predecessor(root).Val
			root.Left = deleteNode(root.Left, root.Val)
		}
	}
	return root
}
func predecessor(root *TreeNode) *TreeNode {
	pre := root.Left
	for pre.Right != nil {
		pre = pre.Right
	}
	return pre
}
func successor(root *TreeNode) *TreeNode {
	post := root.Right
	for post.Left != nil {
		post = post.Left
	}
	return post
}