计算三叉搜索树的高度 - 华为OD机试题

题目描述

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

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

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

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

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

输入描述

第一行为一个数N，表示有N个数，1<=N<=10000

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

输出描述

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

示例一

代码语言：javascript

复制

输入
5
5000 2000 5000 8000 1800
输出
3
说明
最终构造出的树如下，高度为3
。

示例二

代码语言：javascript

复制

输入
3
5000 4000 3000
输出
3
说明
最终构造出的树如下，高度为3
。

java题解

题解

代码语言：javascript

复制

模拟题

按题目要求规则直接构造树，然后递归方式获取树的高度即可。

代码语言：javascript

复制

import java.util.Scanner;
/**


@author code5bug

*/

public class Main {

public static void main(String[] args) {

Scanner scanner = new Scanner(System.in);

int n = scanner.nextInt();

Node root = new Node(scanner.nextInt());
 for (int i = 1; i &lt; n; i++) root.insert(scanner.nextInt());

 System.out.println(root.getHeight());

}

}


class Node {

public int val;

public Node left, mid, right;
public Node(int val) {
    this.val = val;
}

/**
 * 新节点插入树中
 *
 * @param nval
 */
public void insert(int nval) {
    if (nval &lt; this.val - 500) {    // 则将数插入节点的左子树
        if (this.left != null) this.left.insert(nval);
        else this.left = new Node(nval);
    } else if (nval &gt; this.val + 500) { // 则将数插入节点的右子树
        if (this.right != null) this.right.insert(nval);
        else this.right = new Node(nval);
    } else {    // 将数插入节点的中子树
        if (this.mid != null) this.mid.insert(nval);
        else this.mid = new Node(nval);
    }
}

/**
 * 获取树的高度
 *
 * @return
 */
public int getHeight() {
    int maxHeight = 0;
    if (this.left != null) maxHeight = Math.max(maxHeight, this.left.getHeight());
    if (this.mid != null) maxHeight = Math.max(maxHeight, this.mid.getHeight());
    if (this.mid != null) maxHeight = Math.max(maxHeight, this.mid.getHeight());
    return maxHeight + 1;
}

}