450Delete Node in a BST

Insert into a Binary Search Tree

题目描述：

给定一个二叉搜索树的根节点 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

解题思路：首先操作分为两步，第一步找到结点，第二步删除结点。刚开始使用非递归方法，找到结点并不难，但是难于无法删除当前结点，因为无法定位到他的父亲节点以及是左还是右，所以这种方法暂时搁浅。于是转而使用递归方法，发现了递归还是可以实现，因为可以在函数里传参数，父亲节点，左右信息，当前结点等，于是大体基本可以完成，要注意的是对于左右子树都要递归，如果要删除的是叶子结点，那么有了父节点及左右信息，就很容易删除，如果非叶子结点，那么就要选择一个来替代它，两个方法，第一个如果有左孩子，那么找到左子树的最大结点，然后替换，并且要注意的就是当前节点一定是左子树的最右结点，那么他可能还有左子树，所以要让值和要找的值的结点的左孩子指向他的左孩子，另一个有右孩子的情况和这个正好反过来，最后返回root根节点即可。

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(!root)return root;
        TreeNode* p,*q;
        if(key==root->val){
            if(!root->left&&!root->right)return NULL;
            if(root->left){
                left_replace(root);
            }
            else{
                right_replace(root);
            }
        }
        if(root->left)delNode(root->left,1,key,root);
        if(root->right)delNode(root->right,0,key,root);
        return root;
    }
    void delNode(TreeNode* cur,int l,int key,TreeNode* parent){
        if(!cur)return;
        TreeNode* p,*q;
        if(cur->val==key){
            //叶子节点
            if(!cur->left&&!cur->right){
                if(l==1)parent->left=NULL;
                else parent->right=NULL;
            }
            //非叶子结点
            else if(cur->left){
                left_replace(cur);
            }
            else{
                right_replace(cur);
            }
            return;
        }
        delNode(cur->left,1,key,cur);
        delNode(cur->right,0,key,cur);
    }
    void left_replace(TreeNode* cur){
        TreeNode* p,*q;
        q=cur;
        p=cur->left;
        while(p->right){
            q=p;
            p=p->right;
        }
        cur->val=p->val;
        if(q==cur)q->left=p->left;
        else q->right=p->left;
    }
    void right_replace(TreeNode* cur){
        TreeNode* p,*q;
        q=cur;
        p=cur->right;
        while(p->left){
            q=p;
            p=p->left;
        }
        cur->val=p->val;
        if(q==cur)q->right=p->right;
        else q->left=p->right;
    }
};