您的位置:首页 > 其它

二叉树平衡检查

2015-09-01 19:56 225 查看
实现一个函数,检查二叉树是否平衡,平衡的定义如下,对于树中的任意一个结点,其两颗子树的高度差不超过1。

给定指向树根结点的指针TreeNode* root,请返回一个bool,代表这棵树是否平衡。

struct TreeNode {
int val;
char data;
struct TreeNode *left;
struct TreeNode *right;
// TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Balance {
public:
void createBiTree(TreeNode *T)
{
char c;
cin >> c;
if('#' == c)
T = NULL;
else
{
T =new TreeNode();
T->data = c;
createBiTree(T->left);
createBiTree(T->right);
}
}
int deep(TreeNode * T)
{
if(T==NULL) return 0;
int left = deep(T->left);
int right = deep(T->right);
return left>right?left+1:right+1;
}

bool isBalance(TreeNode* root) {
// write code here
if (root==NULL)
{
return true;
}
int left=deep(root->left);
int right= deep(root->right);
if (abs(left-right)>1)
{
return false;
}else
{
return isBalance(root->left)&&isBalance(root->right);	// 返回左子树和右子树的深度之差
}
}
};
void  main()
{
Balance A;
TreeNode *T=NULL;
A.createBiTree(T);
cout<<A.isBalance(T)<<endl;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: