平衡二叉树
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
在这里,我们只需要考虑其平衡性,不需要考虑其是不是排序二叉树。
题解:
1)最长最短法:
根据二叉树的深度来进行判断。找出二叉树的最长路径和最短路径,判断其差值是否小于等于1。
代码如下:
class Solution {
public:
int mi(TreeNode* pRoot){
if(!pRoot) return 0;
int left=ma(pRoot->left);
int right=ma(pRoot->right);
return min(left,right)+1;
}
int ma(TreeNode* pRoot){
if(!pRoot) return 0;
int left=ma(pRoot->left);
int right=ma(pRoot->right);
return max(left,right)+1;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
if(!pRoot) return true;
int len1=ma(pRoot);
int len2=mi(pRoot);
if(abs(len1-len2)<=1)
return true;
else
return false;
}
};
2)一次遍历法:
通过后续遍历二叉树
代码如下:
class Solution {
public:
bool find(TreeNode* pRoot,int &depth){
if(!pRoot){
depth=0;
return true;
}
int left,right;
if(find(pRoot->left,left)&&find(pRoot->right,right)){
if(abs(left-right)<=1){
depth=1+(left>right?left:right);
return true;
}
}
return false;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
int depth=0;
return find(pRoot,depth);
}
};