Obter Fator de Balanceamento

Lição 6 de 16 do curso Árvore AVL - Série de Estruturas de Dados #10 da Coddy.

O fator de balanceamento de um nó é getHeight(node.left) - getHeight(node.right). Um número positivo significa que o lado esquerdo é mais alto, negativo significa que o lado direito é mais alto, e 0 significa que eles estão equilibrados.

Um nó é considerado balanceado desde que seu fator de balanceamento seja -1, 0 ou 1. Qualquer coisa fora desse intervalo (2 ou mais, ou -2 ou menos) é o que desencadeia uma rotação.

Desafio

Iniciante

Escreva um método getBalance(node) em AVLTree que retorna 0 se node for nulo, caso contrário, retorna getHeight(node.left) - getHeight(node.right).

Experimente você mesmo

#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;
}