华为OD机试D卷C卷 - 计算三叉搜索树的高度（C++ Java JavaScript Python C语言）

题目描述

定义构造三叉搜索树规则如下：

每个节点都存有一个数，当插入一个新的数时，从根节点向下寻找，直到找到一个合适的空节点插入。查找的规则是：

• 如果数小于节点的数减去500，则将数插入节点的左子树

• 如果数大于节点的数加上500，则将数插入节点的右子树

• 否则，将数插入节点的中子树

给你一系列数，请按以上规则，按顺序将数插入树中，构建出一棵三叉搜索树，最后输出树的高度。

输入描述

第一行为一个数 N，表示有 N 个数，1 ≤ N ≤ 10000

第二行为 N 个空格分隔的整数，每个数的范围为[1,10000]

输出描述

输出树的高度（根节点的高度为1）

用例1

输入

5
5000 2000 5000 8000 1800

输出

3

说明
最终构造出的树如下，高度为3：

用例2

输入

3
5000 4000 3000

输出

3

说明
最终构造出的树如下，高度为3：

用例3

输入

9
5000 2000 5000 8000 1800 7500 4500 1400 8100

输出

4

说明
最终构造出的树如下，高度为4：

解题思路

定义树类Tree，包含两个方法：insert和getHeight。

1. insert方法用于向树中插入新值。它接受当前树的根节点和要插入的值作为参数。

   • 如果当前节点为空，创建一个新的TreeNode实例，并返回它作为新的根节点。
   • 如果要插入的值小于当前节点值减去500，递归地在左子树中插入该值。
   • 如果要插入的值大于当前节点值加上500，递归地在右子树中插入该值。
   • 如果要插入的值在当前节点值加减500的范围内，递归地在中间子树中插入该值。