一、题目
题目:450. 删除二叉搜索树中的节点
难度:中等
地址:https://leetcode-cn.com/problems/delete-node-in-a-bst/
使用语言:Java
解法关键词:递归、BST搜索+删除
二、代码
针对BST的删除操作需要考虑到:删除时不能破坏BST的特性
/**
* 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;
if(root.val == key){
// 找到目标节点 进行删除操作
// 这两个if把情况1和2都正确处理了
if(root.left == null) return root.right;
if(root.right == null) return root.left;
// 针对第三种情况进行处理
TreeNode minNode = getMin(root.right);
root.val = minNode.val;
root.right = deleteNode(root.right,minNode.val);
}else if(root.val > key){
root.left = deleteNode(root.left,key);
}else if(root.val < key){
root.right = deleteNode(root.right,key);
}
return root;
}
TreeNode getMin(TreeNode node){
// BST最左边的就是最小的
while(node.left != null){
node = node.left;
}
return node;
}
}