Пример #1
0
while True:
    clock.tick(30)
    screen.fill((255, 255, 255))
    for i in pygame.event.get():
        if i.type == pygame.QUIT:
            sys.exit()
        if i.type == pygame.KEYDOWN:
            if i.key == pygame.K_ESCAPE:
                sys.exit()
            if i.key == pygame.K_i:
                print("Enter key(must be number): ")
                key = float(input())
                print("Enter value: ")
                value = input()
                T.insert(Node(key, value))
            if i.key == pygame.K_f:
                if current != None:
                    current.looking = False
                print("Enter key(must be number): ")
                key = float(input())
                current = T.find(key)
                if current != None:
                    current.looking = True
            if i.key == pygame.K_r:
                print("Enter key(must be number): ")
                T.remove(float(input()))
            if i.key == pygame.K_p:
                T.printInOrder()
    drawBT(T.root)
    pygame.display.flip()