题目:
Given a binary tree, determine if it is height-balanced.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
判断一棵树是否是平衡树 每个结点的左右子树高度相差最多为1
Attention:
1. 一旦发现不平衡,递归切断(前后切断);
2. 理解递归的含义,计算当前节点的子树高度(左右子树的判断一定会放到同一个递归中)。递归虽然是调用相同的函数,但是其实都是副本,即使用不同的函数实现。
AC Code:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isBalanced(TreeNode *root) {
//判断一棵树是否是平衡树 每个结点的左右子树高度相差最多为1
//思路:递归判断每个结点。
//Attention: 一旦发现不平衡,递归切断(前后切断);理解递归的含义,计算当前节点的子树高度(左右子树的判断一定会放到同一个递归中)。递归虽然是调用相同的函数,但是其实都是副本,即时不同的函数实现。
bool result = true;
isBalanced_helper(root, result);
return result;
}
private:
int isBalanced_helper(TreeNode* root, bool& result){
if(root == NULL) return 0;
//切断递归,一旦发现树不平衡,结束递归。
if(result == false) return 0;
int leftheight = isBalanced_helper(root->left, result);
int rightheight = isBalanced_helper(root->right, result);
// prevent overwrite of result 同时也是short-circuit
if (result == false) return 0;
if(abs(leftheight - rightheight) > 1) result = false;
//计算当前节点的子树高度。
return max(leftheight, rightheight) + 1;
}
};