华为od机试真题： 计算三叉搜索树的高度(Python)

定义构造三又搜索树规则如下:
每个节点都存有一个数，当插入一个新的数时，从根节点向下寻找，直到找到一个合适的空节点插入查找的规则是:

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

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

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

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

输入描述

第一行为一个数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

题解

1. 节点定义：定义一个包含三个子节点（左、中、右）的节点类 Node。

2. 树的构建

   ：按照题目给出的规则，将一系列数依次插入到树中。规则如下：

   • 如果数小于节点的数减去500，则插入左子树。
   • 如果数大于节点的数加上500，则插入右子树。
   • 否则，插入中子树。

3. 树的高度计算：递归计算树的高度，即根节点的高度为1，叶节点为0。

Python

class Node:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.mid = None
        self.right = None

def insert(root, value):
    if root is None:
        return Node(value)
    
    if value < root.val - 500:
        root.left = insert(root.left, value)
    elif value > root.val + 500:
        root.right = insert(root.right, value)
    else:
        root.mid = insert(root.mid, value)
    
    return root

def height(node):
    if node is None:
        return 0
    left_height = height(node.left)
    mid_height = height(node.mid)
    right_height = height(node.right)
    return max(left_height, mid_height, right_height) + 1

# 输入处理
n = int(input())
values = list(map(int, input().split()))

# 构建三叉树
root = None
for value in values:
    root = insert(root, value)

# 计算并输出树的高度
print(height(root))

2024华为OD机试（C卷+D卷）最新题库【超值优惠】Java/Python/C++合集

🙏整理题解不易， 如果有帮助到您，请给点个赞 ‍❤️‍ 和收藏 ⭐，让更多的人看到。🙏🙏🙏