leetcode 450 删除二叉搜索树中的节点（递归）

https://leetcode-cn.com/problems/delete-node-in-a-bst/

题目

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

1. 首先找到需要删除的节点；
2. 如果找到了，删除它。

• 说明： 要求算法时间复杂度为 O(h)，h 为树的高度。

示例

思路

用递归的方法

1. 如果当前节点不是要删除的节点，根据二叉搜索树的性质往孩子节点去找
2. 如果是，分为几种情况
   (1) 当前节点没有孩子，直接delete， return NULL即可
   (2)当前节点仅有一边孩子，那么将将root的孩子赋给root，再删掉root即可
   (3) 有两边孩子，那么我们有两种解决方案，因为为了保持原有性质，root -> val要大于左边所有，小于右边所有，那么我们可以找到左边的最大值或者右边的最小值，将root -> val赋值成这个值，再根据你是在哪一边找的，递归地删掉这个值即可

/**
 * 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:
    int getNew(TreeNode* root) {
        for (; root -> left; root = root -> left) ;
        return root -> val;
    }
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (!root) return root;
        if (root -> val < key) root -> right = deleteNode(root -> right, key);
        if (root -> val > key) root -> left = deleteNode(root -> left, key);
        if (root -> val == key) {
            if (!root -> left && !root -> right) {
                delete root;
                return NULL;
            }
                
            if (!(root -> left)) {
                TreeNode* p = root -> right;
                delete root; return p;
            }
                
            if (!(root -> right)) {
                TreeNode* p = root -> left;
                delete root; return p;
            }
            if (root -> left && root -> right) {
                int k = getNew(root -> right);
                root -> val = k;
               root -> right = deleteNode(root -> right, k);
            }
        }
        return root;
    }
};