计算二叉树中的平衡因子,并返回不平衡的个数,同时计算平衡二叉树的高度

计算平衡因子并返回不平衡的个数:先序遍历二叉树,计算左右子树的深度,再利用左子树减右子树作为该结点的平衡因子;如果该结点平衡因子大于等于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;
}

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值