计算三叉搜索树的高度

题目
定义构造三叉搜索树规则如下:
每个节点都存有一个数，当插入一个新的数时，从根节点向下寻找，直到找到一个合适的空节点插入。查找的规则是:
如果数小于节点的数减去500，则将数插入节点的左子树
如果数大于节点的数加上500，则将数插入节点的右子树
否则，将数插入节点的中子树
给你一系列数，请按以上规则，按顺序将数插入树中，构建出一棵三叉搜索树，最后输出树的高度。
输入描述
第一行为一个数N，表示有N个数，1≤N≤10000
第二行为N个空格分隔的整数，每个数的范围为[1,10000]
输出描述
输出树的高度(根节点的高度为1)
用例
输入    
5
5000 2000 5000 8000 1800
输出    
3
 

class TrieNode:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.middle = None
        self.right = None

def insert(node, value):
    if value < node.value - 500:
        if node.left is None:
            node.left = TrieNode(value)
        else:
            insert(node.left, value)
    elif value > node.value + 500:
        if node.right is None:
            node.right = TrieNode(value)
        else:
            insert(node.right, value)
    else:
        if node.middle is None:
            node.middle = TrieNode(value)
        else:
            insert(node.middle, value)

def height(node):
    if node is None:
        return 0
    return 1 + max(height(node.left), height(node.middle), height(node.right))

# 用例测试
N = 5
values = [5000, 2000, 5000, 8000, 1800]
root = TrieNode(values[0])
for value in values[1:]:
    insert(root, value)

height(root)