这个非常有技巧,考虑定义这个删除函数以及返回值如下:
删除root->val = key 的节点,并返回新的根节点,很容易想到下面递归代码
if(root==NULL) return root;
if(root->val<key) root->right = deleteNode(root->right,key);
else if(root->val>key) root->left = deleteNode(root->left,key);
如果此时root->val == key
那么有以下几种情况
第一种情况,删除5,没有左子树,直接返回root->right
第二种情况,删除5,没有右子树,直接返回root->left
第三种情况,找到右子数最左边的树,然后将root->left挂上去,返回root->right
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
// 删除root-val=key的root节点,返回值是新的根节点
if(root==NULL) return root;
if(root->val<key) root->right = deleteNode(root->right,key);
else if(root->val>key) root->left = deleteNode(root->left,key);
else{
if(root->left==NULL) return root->right;
else if(root->right==NULL) return root->left;
else{
TreeNode* node = root->right;
while(node->left!=NULL){
node = node->left;
}
node->left = root->left;
return root->right;
}
}
return root;
}
};