reverte height of AVL tree - c

Question

I'm trying to implement an AVL tree for the first time, and apparently it worked, but I need to change the way of marking the height of the tree, the root node is getting the highest height, but I want to change the root node to have height 0 and go incrementing for the child nodes.
#include <stdio.h>
#include <stdlib.h>
struct NO{
int info;
int altura;
struct NO *esq;
struct NO *dir;
};
typedef struct NO* ArvAVL;
ArvAVL* cria_ArvAVL(){
ArvAVL* raiz = (ArvAVL*) malloc(sizeof(ArvAVL));
if(raiz != NULL)
*raiz = NULL;
return raiz;
}
void libera_NO(struct NO* no){
if(no == NULL)
return;
libera_NO(no->esq);
libera_NO(no->dir);
free(no);
no = NULL;
}
void libera_ArvAVL(ArvAVL* raiz){
if(raiz == NULL)
return;
libera_NO(*raiz);//libera cada nó
free(raiz);//libera a raiz
}
int altura_NO(struct NO* no){
if(no == NULL)
return -1;
else
return no->altura;
}
int fatorBalanceamento_NO(struct NO* no){
return labs(altura_NO(no->esq) - altura_NO(no->dir));
}
int maior(int x, int y){
if(x > y)
return x;
else
return y;
}
void emOrdem_ArvAVL(ArvAVL *raiz){
if(raiz == NULL)
return;
if(*raiz != NULL){
emOrdem_ArvAVL(&((*raiz)->esq));
//printf("%d\n",(*raiz)->info);
printf("%d,%d\n",(*raiz)->info,altura_NO(*raiz));
emOrdem_ArvAVL(&((*raiz)->dir));
}
}
void RotacaoLL(ArvAVL *A){
printf("RotacaoLL\n");
struct NO *B;
B = (*A)->esq;
(*A)->esq = B->dir;
B->dir = *A;
(*A)->altura = maior(altura_NO((*A)->esq),altura_NO((*A)->dir)) + 1;
B->altura = maior(altura_NO(B->esq),(*A)->altura) + 1;
*A = B;
}
void RotacaoRR(ArvAVL *A){
printf("RotacaoRR\n");
struct NO *B;
B = (*A)->dir;
(*A)->dir = B->esq;
B->esq = (*A);
(*A)->altura = maior(altura_NO((*A)->esq),altura_NO((*A)->dir)) + 1;
B->altura = maior(altura_NO(B->dir),(*A)->altura) + 1;
(*A) = B;
}
void RotacaoLR(ArvAVL *A){
RotacaoRR(&(*A)->esq);
RotacaoLL(A);
}
void RotacaoRL(ArvAVL *A){
RotacaoLL(&(*A)->dir);
RotacaoRR(A);
}
int insere_ArvAVL(ArvAVL *raiz, int valor){
int res;
if(*raiz == NULL){
struct NO *novo;
novo = (struct NO*)malloc(sizeof(struct NO));
if(novo == NULL)
return 0;
novo->info = valor;
novo->altura = 0;
novo->esq = NULL;
novo->dir = NULL;
*raiz = novo;
return 1;
}
struct NO *atual = *raiz;
if(valor < atual->info){
if((res = insere_ArvAVL(&(atual->esq), valor)) == 1){
if(fatorBalanceamento_NO(atual) >= 2){
if(valor < (*raiz)->esq->info ){
RotacaoLL(raiz);
}else{
RotacaoLR(raiz);
}
}
}
}else{
if(valor > atual->info){
if((res = insere_ArvAVL(&(atual->dir), valor)) == 1){
if(fatorBalanceamento_NO(atual) >= 2){
if((*raiz)->dir->info < valor){
RotacaoRR(raiz);
}else{
RotacaoRL(raiz);
}
}
}
}else{
printf("Valor duplicado!!\n");
return 0;
}
}
atual->altura = maior(altura_NO(atual->esq),altura_NO(atual->dir)) + 1;
return res;
}
struct NO* procuraMenor(struct NO* atual){
struct NO *no1 = atual;
struct NO *no2 = atual->esq;
while(no2 != NULL){
no1 = no2;
no2 = no2->esq;
}
return no1;
}
int remove_ArvAVL(ArvAVL *raiz, int valor){
if(*raiz == NULL){
printf("valor não existe!!\n");
return 0;
}
int res;
if(valor < (*raiz)->info){
if((res = remove_ArvAVL(&(*raiz)->esq,valor)) == 1){
if(fatorBalanceamento_NO(*raiz) >= 2){
if(altura_NO((*raiz)->dir->esq) >= altura_NO((*raiz)->dir->dir))
RotacaoRR(raiz);
else
RotacaoRL(raiz);
}
}
}
if((*raiz)->info < valor){
if((res = remove_ArvAVL(&(*raiz)->dir, valor)) == 1){
if(fatorBalanceamento_NO(*raiz) >= 2){
if(altura_NO((*raiz)->esq->dir) >= altura_NO((*raiz)->esq->esq) )
RotacaoLL(raiz);
else
RotacaoLR(raiz);
}
}
}
if((*raiz)->info == valor){
if(((*raiz)->esq == NULL || (*raiz)->dir == NULL)){
struct NO *oldNode = (*raiz);
if((*raiz)->esq != NULL)
*raiz = (*raiz)->esq;
else
*raiz = (*raiz)->dir;
free(oldNode);
}else {
struct NO* temp = procuraMenor((*raiz)->dir);
(*raiz)->info = temp->info;
remove_ArvAVL(&(*raiz)->dir, (*raiz)->info);
if(fatorBalanceamento_NO(*raiz) <= -2){
if(altura_NO((*raiz)->esq->dir) >= altura_NO((*raiz)->esq->esq))
RotacaoLL(raiz);
else
RotacaoLR(raiz);
}
}
if (*raiz != NULL)
(*raiz)->altura = maior(altura_NO((*raiz)->esq),altura_NO((*raiz)->dir)) + 1;
return 1;
}
(*raiz)->altura = maior(altura_NO((*raiz)->esq),altura_NO((*raiz)->dir)) + 1;
return res;
}
int main()
{
ArvAVL* avl;
int res,i;
int N = 7, dados[7] = {4,5,1,2,3,7,6};
avl = cria_ArvAVL();
for(i=0;i<N;i++){
res = insere_ArvAVL(avl,dados[i]);
}
printf("\nAVL tree:\n");
emOrdem_ArvAVL(avl);
printf("\n\n");
remove_ArvAVL(avl,6);
printf("\nAVL tree:\n");
emOrdem_ArvAVL(avl);
printf("\n\n");
remove_ArvAVL(avl,7);
printf("\nAVL tree:\n");
emOrdem_ArvAVL(avl);
printf("\n\n");
remove_ArvAVL(avl,4);
printf("\nAVL tree:\n");
emOrdem_ArvAVL(avl);
printf("\n\n");
libera_ArvAVL(avl);
return 0;
}
Current output
1,0
2,1
3,0
4,2
5,0
6,1
7,0
Expected output
1,2
2,1
3,2
4,0
5,2
6,1
7,2
AVL Tree
4
/ \
2 6
/ \ / \
1 3 5 7