题目介绍
题目给定一个二叉搜索树的根节点 root 和一个值 key。
要求删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。
一般来说,删除节点可分为两个步骤:
- 首先找到需要删除的节点;
- 如果找到了,删除它。
其次要求算法时间复杂度为 O(h),h 为树的高度。
解题思路
如何找到需要删除的节点?
根据题目要求,我们肯定是要从上到下进行搜索的。其次,我们也要充分利用BST的性质,可以在每轮递归中比较key和根节点的大小,决定到左子树还是右子树里搜寻(这些技巧我们在LeetCode 236中也用到过)。
如何进行删除?
删除节点主要有三种情况:
- 节点只存在左子树,那么我们直接用左子树代替根节点即可;
- 节点只存在右子树,同样地,我们直接用右子树代替根节点;
- 同时存在左右子树是比较复杂的情况,这是我们重点关注的情况。
这里我们优先采用右子树替换根节点的原则