删除二叉搜索树中的节点
- leetcode450. 删除二叉搜索树中的节点
- 题目描述
- 解题思路
- 代码演示
- 二叉树专题
leetcode450. 删除二叉搜索树中的节点
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/delete-node-in-a-bst
题目描述
给定一个二叉搜索树的根节点 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]。
示例 2:
输入: root = [5,3,6,2,4,null,7], key = 0
输出: [5,3,6,2,4,null,7]
解释: 二叉树不包含值为 0 的节点
示例 3:
输入: root = [], key = 0
输出: []
提示:
节点数的范围 [0, 104].
-105 <= Node.val <= 105
节点值唯一
root 是合法的二叉搜索树
-105 <= key <= 105
进阶: 要求算法时间复杂度为 O(h),h 为树的高度。
解题思路
本题总的来说分为两步,第一步是找到要删除的节点,
这个之前有一篇文章说过:二叉搜索树中的搜索
再就是找到后删除节点后,如何调整树的结构,使其还满足二叉搜索树的特性。
这里删除时,就要考虑三种情况了:
1.当前节点没有左右子节点,那就直接删除,
2.左右节点只有一个时,那就直接返回另外一个就行了。
3.第三种左右节点都有时。
我们要把右树上的最小节点替换过来,使其依然满足搜索二叉树的性质。
代码演示
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if(root == null){
return null;
}
return process(root,key);
}
/**
* 递归删除一个节点
* @param root
* @param val
*/
public static TreeNode process(TreeNode root,int val){
if (root == null){
return null;
}
if(root.val == val){
//没有子节点,直接返回null 干掉当前节点
if(root.left == null && root.right == null){
return null;
}
// 左为null 时 返回右节点
if (root.left == null){
return root.right;
}
// 右为null 时 返回左节点
if (root.right == null){
return root.left;
}
//左右节点都有时
//获取右子树上最小的节点
TreeNode min = getMin(root.right);
//删除右子树上最小的节点
root.right = process(root.right,min.val);
//最小节点替换掉要删除的头节点
min.left = root.left;
min.right = root.right;
root = min;
//本题中也可以直接值替换
// root.val = min.val;
} else if (root.val > val) {
root.left = process(root.left,val);
} else if (root.val < val) {
root.right = process(root.right,val);
}
return root;
}
/**
* 获得最小节点,其实就是右树的最左节点
*/
public static TreeNode getMin(TreeNode root){
while (root.left != null){
root = root.left;
}
return root;
}
}
二叉树专题
leetcode701. 二叉搜索树中的插入操作
leetcode700. 二叉搜索树中的搜索
leetcode98. 验证二叉搜索树