# 第四章-平衡二叉树

#include <stdio.h>
#include <stdlib.h>

typedef struct TNode {
int data;
struct TNode *left;
struct TNode *right;
int height;
}TNode,*Tree;

typedef struct QNode {
Tree data;
int top;
int base;
int maxsize;
}QNode,*Queue;

void InitQueue (Queue *q);//初始化队列。
void En (Queue q, Tree T);//进队。
Tree De (Queue q);//出队。
int IsFull (Queue q);//判断是否为满。
int IsEmpty (Queue q);//判断是否为空。
void Level (Tree T);//层次遍历二叉树。

void Level (Tree T)//层次遍历是用来检验二叉树是否平衡正确。
{
Queue q;
InitQueue (&q);
En(q,T);
Tree temp;
while(!IsEmpty(q)) {
temp = De(q);
printf("%d ",temp->data);
if(temp->left) En(q,temp->left);
if(temp->right) En(q,temp->right);
}
}

void InitQueue (Queue *q)
{
(*q) = (QNode*) malloc(sizeof(QNode));
(*q)->top = (*q)->base = 0;
(*q)->maxsize = 10;
(*q)->data = (Tree) malloc(sizeof(TNode) * (*q)->maxsize);
}

int IsFull(Queue q)
{
return ((q->top + 1) % (q->maxsize - 1) == q->base);
}

int IsEmpty (Queue q)
{
return (q->base == q->top);
}

void En (Queue q, Tree T)
{
if (IsFull(q)) {
printf("The queue is full!\n");
}else {
TNode *temp;
temp = T;
q->data[q->top] = *temp;
q->top = (q->top + 1) % (q->maxsize -1 );

}
}

Tree De (Queue q)
{
if(IsEmpty(q)) {
printf("The queue is empty!\n");
}else {
Tree T;
T = &(q->data[q->base]);
q->base = (q->base + 1) % (q->maxsize - 1);
return T;
}
}

int Height (Tree T)//返回当前结点的高度。
{
if(!T) {
return -1;
}else {
return T->height;
}
}

int Max (int a, int b)//返回a，b中最大的数字
{
return a>b?a:b;
}

Tree RotateWithLeft (Tree T)//左旋转来平衡该结点。
{
Tree temp;
temp = T->left;
T->left = temp->right;
temp->right = T;

T->height = Max(Height(T->left),Height(T->right)) + 1;
temp->height = Max (Height(temp->left),Height(temp->right)) + 1;

return temp;
}

Tree RotateWithRight (Tree T)//右旋转平衡该结点。
{
Tree temp;
temp = T->right;
T->right = temp->left;
temp->left = T;

T->height = Max(Height(T->left),Height(T->right)) + 1;//更新T的高度。
temp->height = Max(Height(temp->left),Height(temp->right)) + 1;//更系temp结点的高度。

return temp;
}

Tree RotateWithLeftRight (Tree T)//左右旋转平衡。
{
T->left = RotateWithRight(T->left);

return (RotateWithLeft(T));
}

Tree RotateWithRightLeft (Tree T)//RL旋转平衡。
{
T->right = RotateWithLeft (T->right);

return RotateWithRight(T);
}

Tree Insert (int x, Tree T)
{
if (!T) {//若结点为空，则生成新结点。
T = malloc(sizeof(TNode));
T->data = x;
T->height = 0;
T->left = T->right = NULL;
}else if (x < T->data) {//小于的话，则插入左子树。
T->left = Insert (x,T->left);
if (Height(T->left) - Height(T->right) == 2) {//插入后，若左右子树的高度差等于2，则进行平衡。
if (x < T->left->data) {//若小于，肯定属于LL旋转。
T = RotateWithLeft (T);
}else {//否则属于LR旋转。
T = RotateWithLeftRight(T);
}
}
}else if (x > T->data) {//大于就插入右子树。
T->right = Insert (x,T->right);
if (Height(T->right) - Height(T->left) == 2){
if (x > T->right->data) {//若大于，同上。
T = RotateWithRight(T);
}else {
T = RotateWithRightLeft(T);//同上。
}
}
}

T->height = Max(Height(T->left),Height(T->right)) + 1;//完成后，我们需要更新该结点的高度。

return T;

}

int main ()//测试函数。
{
Tree T;
int i;
for( i =1 ;i < 10;i++) {
T = Insert (i,T);
//printf("data is %d",T->data);
}

Level (T);

return 0;

}

posted @ 2020-10-19 21:46  lookfd  阅读(29)  评论(0编辑  收藏