450.go

package main

//给定一个二叉搜索树的根节点 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
//delete-node-in-a-bs
func deleteNodeInBS(root *TreeNode, key int) *TreeNode {
	has, node := findNodeInBS(root, key)
	if has {
		if node.Left == nil {
			return node.Right
		}
		if node.Right == nil {
			return node.Left
		}

		min := node.Right
		for min.Left != nil {
			min = min.Left
		}

		min.Left = node.Left
		root = root.Right
	}
	return root
}

func findNodeInBS(root *TreeNode, key int) (bool, *TreeNode) {
	if root == nil {
		return false, nil
	}
	if root.Val == key {
		return true, root
	} else if root.Val < key {
		return findNodeInBS(root.Left, key)
	} else if root.Val > key {
		return findNodeInBS(root.Right, key)
	}
	return false, nil
}