题目描述:
代码如下:
/**
* 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(key==root.val){
//分三种情况
//1、左右结点都为空 直接删掉
//2、子树有一边为空,左(右)子结点代替父结点
//3、子树都不为空,将父亲结点与左子树的最大或者右子树的最小结点交换 然后删掉
if(root.left==null){
return root.right;
}
if(root.right==null){
return root.left;
}
//这里用右子树的最小结点
TreeNode min=getMin(root.right);
root.val=min.val;
root.right=deleteNode(root.right,min.val);//复制上来之后 从右边子树删除掉
//涉及更改树的操作,需要对递归调用的返回值进行接收
}else if(key<root.val){
root.left=deleteNode(root.left,key);
}else if(key>root.val){
root.right=deleteNode(root.right,key);
}
return root;
}
TreeNode getMin(TreeNode root){
if(root.left==null){
return root;
}
return getMin(root.left);
}
}