Получение коэффициента баланса

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

Коэффициент сбалансированности узла равен getHeight(node.left) - getHeight(node.right). Положительное число означает, что левая сторона выше, отрицательное — что правая сторона выше, а 0 означает, что они равны.

Узел считается сбалансированным, пока его коэффициент сбалансированности равен -1, 0 или 1. Любое значение вне этого диапазона (2 или более, или -2 или менее) вызывает поворот.

Задание

Новичок

Напишите метод getBalance(node) в классе AVLTree, который возвращает 0, если node равен null, в противном случае возвращает getHeight(node.left) - getHeight(node.right).

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

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

static Node* mkNodeWithHeight(int h) {
    Node* n = Node_create(0);
    n->height = h;
    return n;
}

int main(void) {
    AVLTree* tree = AVLTree_create();
    char line[256];
    while (fgets(line, sizeof(line), stdin) != NULL) {
        line[strcspn(line, "\r\n")] = '\0';
        char lTok[64];
        char rTok[64];
        sscanf(line, "%63s %63s", lTok, rTok);
        Node* root = Node_create(0);
        root->left = NULL;
        if (strcmp(lTok, "null") != 0) {
            root->left = mkNodeWithHeight(atoi(lTok));
        }
        root->right = NULL;
        if (strcmp(rTok, "null") != 0) {
            root->right = mkNodeWithHeight(atoi(rTok));
        }
        printf("%d\n", AVLTree_getBalance(tree, root));
    }
    return 0;
}