华为OD机试 Python【计算三叉搜索树的高度】

题目

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

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

如果数小于节点的数减去500，则将数插入节点的左子树
如果数大于节点的数加上500，则将数插入节点的右子树
否则，将数插入节点的中子树
给你一系列数，请按以上规则，按顺序将数插入树中，构建出一棵三叉搜索树，最后输出树的高度。

输入描述
第一行为一个数 N，表示有 N 个数，1 ≤ N ≤ 10000
第二行为 N 个空格分隔的整数，每个数的范围为[1,10000]

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

用例

输入	
5
5000 2000 5000 8000 1800
输出	
3

解题思路

首先，定义一个树节点类 TreeNode，它包含一个整数值和三个子节点（左、中、右）。然后按照规则逐个插入每个数值到树中。在插入过程中，根据规则选择将数值插入到节点的左、中、右子树中。同时，我们在插入过程中记录树的高度，即从根节点到最深叶子节点的距离。最后输出树的高度。整体思路是构建一个三叉