题目描述
实现一个函数,检查二叉树是否平衡,平衡的定义如下,对于树中的任意一个结点,其两颗子树的高度差不超过1。
给定指向树根结点的指针TreeNode* root,请返回一个bool,代表这棵树是否平衡
//方法1
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Balance {
public:
int getDepth(TreeNode* root)
{
if(root==NULL)
return 0;
int leftDepth=getDepth(root->left);
int rightDepth=getDepth(root->right);
return leftDepth>rightDepth?(leftDepth+1):(rightDepth+1);
}
bool isBalance(TreeNode* root)
{
if(root==NULL)
return true;
int leftDepth=getDepth(root->left);
int rightDepth=getDepth(root->right);
int diff=leftDepth-rightDepth;
if(diff>1||diff<-1)
return false;
return isBalance(root->left)&&isBalance(root->right);
}
};
上面这种方法虽然代码简洁,但是由于一个结点会被重复遍历多次。这种思路的时间效率不高。
因此,如果我们后序遍历的方式遍历二叉树的每一个结点,在遍历一个结点之前我们就已经知道了它的左右字树。只要在遍历的时候记录它的深度(某一结点的深度等于它到叶子结点的路径长度),这样就可以一边遍历一边判断每个结点是不是平衡的 。
class Balance
{
public:
bool isBalance(TreeNode* root,int* Depth)
{
if(root==NULL)
{
*Depth=0;
return true;
}
int left,right;
if(isBalance(root->left,&left)&&isBalance(root->right,&right))
{
int diff=left-right;
if(diff<=1&&diff>=-1)
{
*Depth=1+(left>right)?left:right;
return true;
}
}
return false;
}
bool isBalance(TreeNode* root)
{
int depth=0;
return isBalance(root,&depth);
}
};