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

代码随想录算法训练营 | 博客打卡
题目链接：leetcode 450. 删除二叉搜索树中的节点
注：仅为个人笔记，难以细致且通俗。

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

// 找到待删节点后有三种情况：
// 1、待删节点为叶子节点（没有孩子）
// 2、待删节点为仅有一个孩子节点
// 3、待删节点有两个孩子

// 三种情况如何处理？
// 1、直接删就行，也就是返回nil，让上一层接收个nil就相当于删除了
// 2、让这一个孩子代替当前节点。具体操作是返回这个孩子节点。
// 3、让右孩子代替当前节点，左孩子接到右孩子的最左边，也就是左孩子的后继结点，也是右子树的最小值（或者用一个二叉树通用的方案，直接将待删节点左孩子的后继节点代替待删节点）
func deleteNode(root *TreeNode, key int) *TreeNode {
    switch {
    case root == nil:
        return nil
    case root.Val > key:
        root.Left = deleteNode(root.Left, key)
    case root.Val < key:
        root.Right = deleteNode(root.Right, key)

    // root.Val == key
    // 这一分支可处理两种情况，即上述的情况1与2
    case root.Left == nil || root.Right == nil:
        if root.Left != nil {
            return root.Left
        }
        return root.Right
    // 处理第3种情况
    default:
        // successor 后继节点
        successor := root.Right
        // 寻找待删节点左孩子的后继节点
        for successor.Left != nil {
            successor = successor.Left
        }
        // 将后继结点提升至待删位置
        successor.Right = deleteNode(root.Right, successor.Val)
        successor.Left = root.Left
        return successor
    }
    return root
}