给定一个二叉搜索树的根节点 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
/** * Definition for a binary tree node. * type TreeNode struct { * Val int * Left *TreeNode * Right *TreeNode * } */ func deleteNode(root *TreeNode, key int) *TreeNode { if (root == nil) { return nil } if (key < root.Val) { root.Left = deleteNode(root.Left, key) return root } else if (key > root.Val) { root.Right = deleteNode(root.Right, key) return root } else { if (root.Left == nil) { return root.Right } if (root.Right == nil) { return root.Left } // new 和 make的区别,要记住 //通过make创建对象 make只能创建slice chan map //make 只能用来分配及初始化类型为 slice、map、chan 的数据。new 可以分配任意类型的数据; //new 分配返回的是指针,即类型 *Type。make 返回引用,即 Type; //new 分配的空间被清零。make 分配空间后,会进行初始化; //make 函数只用于 map,slice 和 channel,并且不返回指针 node := new(TreeNode) node = findMin(root.Right) root.Val = node.Val root.Right = deleteNode(root.Right, node.Val) } return root } func findMin(node *TreeNode) *TreeNode { for { // golang 没有while if node.Left == nil { break } node = node.Left } return node }