华为OD机试统一考试D卷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:
Java
import java.util.Scanner;
// 主类
public class Main {
// 静态内部类:树
static class Tree {
// 插入方法:向树中插入值
public TreeNode insert(TreeNode root, int val) {
if (root == null) {
return new TreeNode(val); // 如果根节点为空,创建新节点作为根节点
}
if (val < root.val - 500) {
root.left = insert(root.left, val); // 如果值小于根节点值减500,插入到左子树
} else if (val > root.val + 500) {
root.right = insert(root.right, val); // 如
剩余60%内容,订阅专栏后可继续查看/也可单篇购买
华为OD机试刷题 文章被收录于专栏
华为OD机试刷题记录机考算法题库,帮助你上岸华为。提供C++/Java、JavaScript、Python四种语言的解法。