计算平衡因子并返回不平衡的个数:先序遍历二叉树,计算左右子树的深度,再利用左子树减右子树作为该结点的平衡因子;如果该结点平衡因子大于等于2或者小于等于-2,则为不平衡结点。重复上述步骤,直到最终求得所有不平衡二叉树结点个数。
计算平衡二叉树深度:先序遍历二叉树,每次选择子树高度较大的子树进行计算,最终求得整棵树高度。
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef char ElemType;
typedef struct BSTNode {
ElemType data;
struct BSTNode* lchild;
struct BSTNode* rchild;
int bf;
}BSTNode, * BSTree;
//中序遍历
void inOrder(BSTree T) {
if (T != NULL) {
inOrder(T->lchild);
printf("%c", T->data);
inOrder(T->rchild);
}
}
//插入操作
void insertBSTree(BSTree& T, ElemType num) {
if (T == NULL) {
T = (BSTNode*)malloc(sizeof(BSTNode));
T->data = num;
T->lchild = NULL;
T->rchild = NULL;
}
else if (T->data < num) {
insertBSTree(T->rchild, num);
}
else if (T->data > num) {
insertBSTree(T->lchild, num);
}
}
//创建二叉排序树
BSTree creatBSTree() {
ElemType num;
BSTNode* T = NULL;
while (scanf("%c", &num) && num != '\n') {
insertBSTree(T, num);
}
return T;
}
//计算树的深度
/*算法思想:二叉树中存在两种类型的树,即没有结点的空树和有节点的非空树。空树的深度为0,非空树的深度为左右子树深度最大值加1,即为整棵树的深度*/
int getlength(BSTNode* T) {
if (T == NULL) {//空树,深度为0
return 0;
}
else {//非空树,树的高度为左右子树深度最大者加1
int leftlength = getlength(T->lchild);
int rightlength = getlength(T->rchild);
return leftlength > rightlength ? leftlength + 1 : rightlength + 1;
}
}
//获得平衡因子,并返回不平衡的结点个数
int getBalance(BSTree T) {
if (T == NULL) {
return 0;
}
int leftlength = getlength(T->lchild);
int rightlength = getlength(T->rchild);
T->bf = leftlength - rightlength;
//如果当前结点不平衡,则以当前节点为根的子树中不平衡结点数量为:1+左子树不平衡结点数量+右子树不平衡结点数量
if (T->bf >= 2 || T->bf <= -2) {
return 1 + getBalance(T->lchild) + getBalance(T->rchild);
}
//如果当前结点平衡,则以当前节点为根的子树中不平衡结点数量为:左子树不平衡结点数量+右子树不平衡结点数量
return getBalance(T->lchild) + getBalance(T->rchild);
}
/*算法思想:先序遍历二叉树,每次选择根节点平衡因子较大的子树进行高度计算,最终子树高度为整棵树的高度*/
int getBalancelength(BSTree T) {
if (T == NULL) {
return 0;
}
//平衡因子大于0,则说明左子树的深度大于右子树,直接求左子树深度加1,即为整颗树的深度
return T->bf > 0 ? getBalancelength(T->lchild) + 1 : getBalancelength(T->rchild) + 1;
}
int main() {
BSTree T = creatBSTree();
inOrder(T);
printf("\n");
int x=getBalance(T);
printf("%d\n", x);
int k = getBalancelength(T);
printf("%d\n", k);
return 0;
}