华为OD机试E卷 --计算三叉搜索树的高度--24年OD统一考试（Java & JS & Python & C & C++）

题目描述

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

查找的规则是:

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

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

输入描述

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

输出描述

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

用例

输入

5
5000 2000 5000 8000 1800

输出

3

说明
无

输入

3
5000 4000 3000

输出

3

说明
无

输入

9
5000 2000 5000 8000 1800 7500 4500 1400 8100

输出

4

说明
无

题目解析

1. 定义一个三叉树节点类，包含三个子节点（左、中、右）和一个值。
2. 定义一个插入函数，按照题目给定的规则插入新的节点。
3. 定义一个计算树高度的函数。
4. 读取输入数据，依次插入到树中。
5. 输出树的高度。

JS算法源码

class TriTreeNode {
   
    constructor(val) {
   
        this.val = val;
        this.left = null;
        this.middle = null;
        this.right = null;
    }
}

function insert(root, val) {
   
    if (!root) {
   
        return new TriTreeNode(val);
    }
    if (val < root.val - 500) {
   
        root.left = insert(root.left, val);
    } else if (val > root.val + 500) {
   
        root.right = insert(root.right, val);
    } else {
   
        root.middle = insert(root.middle, val);
    }
    return root;
}

function getHeight(node) {
   
    if (!node) {
   
        return 0;
    }
    return 1 + Math.max(getHeight(node.left), getHeight(node.middle), getHeight(node.right));
}

// 读取输入
const readline = require('readline').createInterface({
   
    input: process.stdin,
    output: process.stdout
});

readline.question('', (N) => {
   
    readline.question('', (line) => {
   
        const nums = line.split(' ').map(Number);
        let root = null;
        for (const num of nums) {