LeetCode_450. 删除二叉搜索树中的节点

题目描述：

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点；
如果找到了，删除它。
说明： 要求算法时间复杂度为 O(h)，h 为树的高度。

示例:

root = [5,3,6,2,4,null,7]
key = 3

    5
   / \
  3   6
 / \   \
2   4   7

给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。

一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。

    5
   / \
  4   6
 /     \
2       7

另一个正确答案是 [5,2,6,null,4,null,7]。

    5
   / \
  2   6
   \   \
    4   7

思路：删除的节点一共分为三种情况
1、删除的节点是叶子节点，则直接删除
2、删除的节点只有一侧子树，则删除节点后，将这侧子树连接到删除节点的位置
3、删除的节点左右子树都存在，则需要找到左子树的最大节点或者右子树的最小节点，然后赋值给当前节点，再去左子树删除这个最大节点或者去右子树删除这个最小节点

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *findMin(TreeNode* root){
        while(root->left)
            root=root->left;
        return root;
    }
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root==NULL)
            return NULL;
        if(key<root->val)//要删除的节点在左子树
            root->left=deleteNode(root->left,key);//到左子树上删除节点
        else if(key>root->val)//要删除的节点在右子树
            root->right=deleteNode(root->right,key);//到右子树上删除节点
        else{//当前root就是要删除的节点
            if(root->left==NULL){
                TreeNode* right=root->right;
                delete root;//要真真切切的删除节点，不能有节点残留
                return right;
            }
            else if(root->right==NULL){
                TreeNode* left=root->left;
                delete root;
                return left;
            }
            //找到左子树中的最大值，赋值给当前根节点，再递归删除左子树的最大值
            //或者找到右子树的最小值，赋值给当前根节点，再递归删除右子树的最小值
            else{
                TreeNode* minNode=findMin(root->right);
                root->val=minNode->val;
                root->right=deleteNode(root->right,root->val);
            }       
        }
        return root;
    }
};