微软高频面试题 Leetcode 450. 删除二叉搜索树中的节点

这个非常有技巧，考虑定义这个删除函数以及返回值如下：

删除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;
    }
};