题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
方法一:比较各节点深度(存在重复遍历节点情况)。
class Solution {
public:
int getDepth(TreeNode* root)
{
if(root == nullptr)
return 0;
int leftH = getDepth(root->left);
int rightH = getDepth(root->right);
return max(leftH, rightH) + 1;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot == nullptr)
return true;
int left = getDepth(pRoot->left);
int right = getDepth(pRoot->right);
int dif = left - right;
if(dif > 1 || dif < -1)
return false;
return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
}
};
方法二:使用后续遍历方式遍历二叉树的每个节点,在遍历到一个节点之前我们就已经遍历了它的左右子树,则不存在重复遍历。
class Solution {
public:
bool isBalanced(TreeNode* root, int& depth)
{
if(root == nullptr)
{
//depth = 0;
return true;
}
int left = 0, right = 0;
if(isBalanced(root->left, left) && isBalanced(root->right, right))
{
int dif = left - right;
if(dif <= 1 && dif >= -1)
{
depth = max(left, right) + 1;
return true;
}
}
return false;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
int depth = 0;
return isBalanced(pRoot, depth);
}
};