Pobieranie współczynnika równowagi

Lekcja 6 z 16 w kursie Drzewo AVL – struktury danych, seria #10 w Coddy.

Współczynnik równowagi węzła to getHeight(node.left) - getHeight(node.right). Wartość dodatnia oznacza, że lewa strona jest wyższa, ujemna — że wyższa jest prawa strona, a 0 oznacza, że obie strony są równe.

Węzeł uznaje się za zrównoważony, jeśli jego współczynnik równowagi wynosi -1, 0 lub 1. Każda wartość spoza tego zakresu (2 lub więcej albo -2 lub mniej) powoduje rotację.

Wyzwanie

Początkujący

Napisz metodę getBalance(node) w klasie AVLTree, która zwraca 0, jeśli node ma wartość null, a w przeciwnym razie zwraca getHeight(node.left) - getHeight(node.right).

Spróbuj swoich sił

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