Правый поворот

Урок 7 из 16 курса AVL-дерево — Серия «Структуры данных» №10 на Coddy.

Когда левая сторона узла слишком высока, это исправляет правый поворот. Назовем несбалансированный узел y, а его левого потомка — x. Поворот продвигает x на место y: x становится новым корнем поддерева, y становится правым потомком x, а старое правое поддерево x (назовем его T2) переприкрепляется как новое левое поддерево y, так как каждое значение в T2 по-прежнему больше x и меньше y.

После перепривязки указателей необходимо пересчитать height для y и x, причем для y в первую очередь, так как теперь он находится ниже в дереве.

Задание

Легко

Напишите метод rotateRight(y) в классе AVLTree. Пусть x = y.left и T2 = x.right. Установите x.right = y и y.left = T2, пересчитайте height для y, а затем для x по формуле 1 + max(getHeight(left), getHeight(right)), и верните x в качестве нового корня поддерева.

Попробуйте сами

#include <stdio.h>
#include <string.h>
#include "avltree.h"

int main(void) {
    AVLTree* tree = AVLTree_create();
    char line[256];
    while (fgets(line, sizeof(line), stdin) != NULL) {
        line[strcspn(line, "\r\n")] = '\0';
        if (strcmp(line, "1") == 0) {
            Node* y = Node_create(30);
            y->height = 3;
            Node* x = Node_create(20);
            x->height = 2;
            Node* t1 = Node_create(10);
            t1->height = 1;
            x->left = t1;
            y->left = x;
            Node* newRoot = AVLTree_rotateRight(tree, y);
            printf("%d %d %d %d\n", newRoot->value, newRoot->height, newRoot->left->value, newRoot->right->value);
        }
        if (strcmp(line, "2") == 0) {
            Node* y = Node_create(50);
            y->height = 3;
            Node* x = Node_create(30);
            x->height = 2;
            x->left = Node_create(20);
            x->left->height = 1;
            x->right = Node_create(40);
            x->right->height = 1;
            y->left = x;
            y->right = Node_create(60);
            y->right->height = 1;
            Node* newRoot = AVLTree_rotateRight(tree, y);
            printf("%d %d %d %d %d %d\n", newRoot->value, newRoot->height, newRoot->left->value, newRoot->right->value, newRoot->right->left->value, newRoot->right->right->value);
        }
    }
    return 0;
}