Beispiel #1
0
def main():
    rb_tree = RedBlackTree()

    # корень
    node_10_value = Autoservice(10, 'Переделкино', 'Вася Пупкин', 'Омск')
    root = Node(value=node_10_value, color=BLACK, parent=None, left=NIL_LEAF, right=NIL_LEAF)
    node_10 = root

    # левое поддерево
    node_5_value = Autoservice(5, 'Сломай меня полностью', 'Иннокентий Пересядько', 'Якутск')
    node_5 = Node(value=node_5_value, color=BLACK, parent=root, left=NIL_LEAF, right=NIL_LEAF)

    # правое поддерево
    node_20_value = Autoservice(20, 'Загони под шины', 'Яков Явенчучечков', 'Ханты-Мансийск')
    node_20 = Node(value=node_20_value, color=RED, parent=root, left=NIL_LEAF, right=NIL_LEAF)
    node_15_value = Autoservice(15, 'Вест Коуст Кастомс', 'Евакий Иксзибитов', 'Бирюлёво')
    node_15 = Node(value=node_15_value, color=BLACK, parent=node_20, left=NIL_LEAF, right=NIL_LEAF)
    node_25_value = Autoservice(25, 'Подкуй братуху', 'Каракантанбек Изумнетралитанаев', 'Владивосток')
    node_25 = Node(value=node_25_value, color=BLACK, parent=node_20, left=NIL_LEAF, right=NIL_LEAF)
    node_20.left = node_15
    node_20.right = node_25

    node_12_value = Autoservice(12, 'Железный сивый мерин', 'Эщкере Капиталистов', 'Староперуново')
    node_12 = Node(value=node_12_value, color=RED, parent=node_15, left=NIL_LEAF, right=NIL_LEAF)
    node_17_value = Autoservice(17, 'Вах Какой Машына', 'Джамшут Равшанов', 'Перезауехово')
    node_17 = Node(value=node_17_value, color=RED, parent=node_15, left=NIL_LEAF, right=NIL_LEAF)
    node_15.left = node_12
    node_15.right = node_17

    root.left = node_5
    root.right = node_20
    rb_tree.root = root

    node_19_value = Autoservice(19, 'На созвоне & вопросики порешаем & тоси-боси', 'Капитан Глазозоркость', 'Готтэм')

    print('Before:\n' + '-' * 80)
    for i in list(rb_tree):
        print(i)

    rb_tree.add(node_19_value)

    """
                ____10B____                           ____10B____
               5B      __20R__                       5B      __20R__
                  __15B__     25B   -- КРАСИМ В -->      __15R__    25B
               12R      17R                           12B      17B
               добавляем-->19R                                   19R

Общее
направление:                   ____10B____
   LR=>RL                   5B         ___15R___
правая ротация                       12B      __20R__
                                            17B      25B
                                              19R


                                 _____15B_____
    Левая ротация к           10R           __20R__
                            5B  12B       17B      25B
                                            19R
    """

    print('After:\n' + '-' * 80)
    for i in list(rb_tree):
        print(i)

    print('Trying to find \n{}\n in tree: \n{}'.format(node_17_value, rb_tree.find_node(node_17_value)))
    print()
    print('Is \n{}\n in tree? {}'.format(node_17_value, rb_tree.contains(node_17_value)))

    print('Tree root has children: {}'.format(rb_tree.root.has_children()))
    print('Tree root has {} children'.format(rb_tree.root.count_children()))

    values = [
        node_10_value,
        node_5_value,
        node_20_value,
        node_15_value,
        node_25_value,
        node_12_value,
        node_17_value
    ]

    print('-' * 80)
    rb_tree = RedBlackTree()
    for value in values:
        rb_tree.add(value)
        for i in list(rb_tree):
            print(i)
        print('-' * 80)

    print('-' * 80)
    for value in values:
        for i in list(rb_tree):
            print(i)
        rb_tree.remove(value)
        print('-' * 80)
Beispiel #2
0
while (number):
    print(
        "Digite:\n1- Para adicionar a árvore.\n2- Para remover da árvore\n0- Para sair\n"
    )
    number = int(input())
    if number == 1:
        print("Digite o valor que queira adicionar.")
        value = int(input())
        tree.add(value)
        root = tree.root
        print_tree(root)

    elif number == 2:
        print("Digite o valor que deseje remover.")
        value = int(input())
        result = tree.remove(value)
        if result:
            print("Valor: {} retirado com sucesso.".format(value))
            root = tree.root
            print_tree(root)
        else:
            print("Valor não se encontra na árvore.")

    print("Aperte enter para continuar!!!")
    try:
        input()
    except:
        print("Você está no python2 por favor aperte enter de novo")
        raw_input()
    os.system('clear')