题目描述:
实现一个函数,检查二叉树是否平衡。在这个问题中,平衡树的定义如下:任意一个节点,其两棵子树的高度差不超过 1。
思路:
暴力解法,想办法遍历到二叉树的每个节点,然后计算每个节点是否平衡,通过实现计算节点高度的函数,计算每个节点的左右子树高度,检查高度差。
但是在遍历二叉树节点过程中,计算高度会重复计算好多次相同节点的高度,导致效率低下,可以在计算高度函数中,每次计算出高度就来判断是否满足高度差,如果不满足,将全局变量设为false,并通过全局变量结束递归即可。
答案:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isBalance = true;
bool isBalanced(TreeNode* root) {
computeDepth(root);
return isBalance;
}
int computeDepth(TreeNode* node){
if(node == nullptr) return 0;
int leftDepth = computeDepth(node->left);
if(!isBalance) return -1;
int rightDepth = computeDepth(node->right);
if(!isBalance) return -1;
if(abs(leftDepth - rightDepth) > 1) isBalance = false;
return max(leftDepth,rightDepth) + 1;
}
};