示例#1
0
def get_avg_height(N: int, t: int, print_statements=False) -> float:
    height_sum = 0
    for i in range(t):
        rand = [random.randrange(1, 501) for x in range(N)]
        tree = Tree()
        for x in rand:
            tree.insert(x)

        height_sum += tree.height()
        if print_statements:
            print("The height of the tree with {} nodes is {}".format(
                N, tree.height()))
    avg = height_sum / t
    if print_statements:
        print("The average height is", avg)
    return avg