判断二叉树是否平衡,计算树的高度

#include <iostream>
using namespace std;
int count = 0;
class BinTree{
public:
	BinTree(int v){
		value = v;
		left = NULL;
		right = NULL;
	}
	int value;
	BinTree* left;
	BinTree* right;
};
int max(int a,int b) {
	return a>b?a:b;
}
void createTree(BinTree* &tree,int depth,int cur) {
	if(cur < depth) {
		tree = new BinTree(count);
		count++;
		createTree(tree->left,depth,cur+1);
		createTree(tree->right,depth,cur+1);
	} else {
		return;
	}
	
}
bool isBalance(BinTree* tree,int &height){
	if(!tree) { 
		height = 0;
		return true;
	}
	int lh,rh;
	if(!isBalance(tree->left,lh))
		return false;
	if(!isBalance(tree->right,rh))
		return false;
	height = max(lh,rh) + 1;
	if(abs(lh-rh) > 1)
		return false;
	return true;
	
}
int main(){

	BinTree* tree = new BinTree(0);
	BinTree* leftTree;
	BinTree* rightTree;
	createTree(leftTree,11,0);
	createTree(rightTree,10,0);
	tree->left = leftTree;
	tree->right = rightTree;
	int height;
	cout<<(isBalance(tree,height)?"true":"false")<<endl;
	cout<<height<<endl;
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值