refactor: remove buildTree (#716)

This commit is contained in:
nuomi1
2023-09-02 19:47:54 +08:00
committed by GitHub
parent fd84d4443e
commit 22d1135e04
@@ -10,9 +10,10 @@ import utils
class BinarySearchTree {
private var root: TreeNode?
init(nums: [Int]) {
let nums = nums.sorted() // 排序数组
root = buildTree(nums: nums, i: 0, j: nums.count - 1) // 构建二叉搜索树
/* 构造方法 */
init() {
// 初始化空树
root = nil
}
/* 获取二叉树根节点 */
@@ -20,20 +21,6 @@ class BinarySearchTree {
root
}
/* 构建二叉搜索树 */
func buildTree(nums: [Int], i: Int, j: Int) -> TreeNode? {
if i > j {
return nil
}
// 将数组中间节点作为根节点
let mid = (i + j) / 2
let root = TreeNode(x: nums[mid])
// 递归建立左子树和右子树
root.left = buildTree(nums: nums, i: i, j: mid - 1)
root.right = buildTree(nums: nums, i: mid + 1, j: j)
return root
}
/* 查找节点 */
func search(num: Int) -> TreeNode? {
var cur = root
@@ -154,8 +141,12 @@ enum _BinarySearchTree {
/* Driver Code */
static func main() {
/* 初始化二叉搜索树 */
let nums = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
let bst = BinarySearchTree(nums: nums)
let bst = BinarySearchTree()
// 请注意,不同的插入顺序会生成不同的二叉树,该序列可以生成一个完美二叉树
let nums = [8, 4, 12, 2, 6, 10, 14, 1, 3, 5, 7, 9, 11, 13, 15]
for num in nums {
bst.insert(num: num)
}
print("\n初始化的二叉树为\n")
PrintUtil.printTree(root: bst.getRoot())