华为OD机试统一考试D卷C卷 - 计算三叉搜索树的高度 py

题目描述

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

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

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

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

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

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

输入描述

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

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

输出描述

输出树的高度(根节点的高度为1)

用例1

输入

5
5000 2000 5000 8000 1800

输出

3

说明 最终构造出的树如下,高度为3:

alt

用例2

输入

3
5000 4000 3000

输出

3

说明 最终构造出的树如下,高度为3:

alt

用例3

输入

9
5000 2000 5000 8000 1800 7500 4500 1400 8100

输出

4

说明 最终构造出的树如下,高度为4:

alt

Python

class TreeNode:
    def __init__(self, val):
        self.val = val  # 节点值
        self.left = self.mid = self.right = None  # 左、中、右子节点

class Tree:
    # 插入方法:向树中插入值
    def insert(self, root, val):
        if root is None:
            return TreeNode(val)  # 如果根节点为空,创建新

剩余60%内容,订阅专栏后可继续查看/也可单篇购买

2024华为OD机试E卷D卷题 文章被收录于专栏

本专栏给大家提供了华为2024最新华为OD(D)卷的题目汇总。华为OD机试刷题记录机考算法题库,帮助你上岸华为。提供C++/Java、JavaScript、Python四种语言的解法。

全部评论
机试换题目了?
点赞 回复 分享
发布于 08-14 22:41 浙江

相关推荐

头像
08-23 22:25
门头沟学院 Java
点赞 评论 收藏
分享
1 1 评论
分享
牛客网
牛客企业服务