leetcode450-删除二叉搜索树中的节点

leetcode 450

思路

对于要删除节点的操作，我们首先需要知道删除的节点有哪几种情况

• 没有要删除的节点
• 删除的节点为叶子节点
• 删除的节点左节点为空 右节点不为空
• 删除的节点做节点不为空 右节点为空
• 删除的节点左右节点都不为空
  我们如果查找到当前节点的值是目标值，那么我们需要判断这些情况，然后进行处理，返回的值会在后面进行接收，比如示例中假设要删除的节点是7，7此时是叶子节点，那么我们就需要返回null, 原本6->left = 7 现在更新后变成 6->left = null
  假设要删除的节点是6，那么我们发现6有右子树，没有左子树，那么我们就返回6的右子树(7)，让6的父节点5指向7
  如果要删除的节点是3，我们发现3的左右节点都有值，那么我们需要把左右节点都重构，因为是二叉搜索树，所以根据特性，我们可以让3的左子树作为子树根节点也可以让3的右子树作为子树根节点，此时我们让3的右子树作为根节点，那么我们知道3的左子树的任何节点都是比3的右子树的任何节点都更小的，所以我们需要找到右子树的最小的节点，也就是一直向左边遍历到底，最后一个元素是右子树最小值，此时再让最小值的左子树为根节点3的左子树就重构完成
  下图是删除操作的过程：可以看出我们先找到两个节点：删除元素7的左子树节点5，删除元素7的右子树内的最小值8，让8->left = 5，返回节点9(7->right)，设置2->right = 9
  
  图片来自代码随想录

实现

var deleteNode = function (root, val) {
    // 没找到删除的元素
    if (!root) return root;
    if (root.val === val) {
        if (!root.left && !root.right) {
            // 叶子节点
            return null
        } else if (root.left && !root.right) {
            // 左节点存在 右节点不存在
            return root.left
        } else if (!root.left && root.right) {
            // 左节点不存在，右节点存在
            return root.right;
        } else if (root.left && root.right) {
            // 左右节点都存在，重构子树
            return getCurNode(root);
        }
    }
    if(val < root.val){
        root.left = deleteNode(root.left,val)
    }

    if(val > root.val){
        root.right = deleteNode(root.right, val)
    }
    return root;
};
// 对于左右节点都存在的情况进行重新构造
var getCurNode = function (root) {
    let curNode = root.right;
    while (curNode.left) curNode = curNode.left;
    curNode.left = root.left;
    return root.right;
}