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

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

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

模拟题

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

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 < 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 < this.val - 500) {    // 则将数插入节点的左子树
            if (this.left != null) this.left.insert(nval);
            else this.left = new Node(nval);
        } else if (nval > 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;
    }

}