提示
LintCode中的相关算法题实现代码,可以在我的GitHub中下载。
题目需求
描述
给定一个二叉树,确定它是高度平衡的。对于这个问题,一棵高度平衡的二叉树的定义是:一棵二叉树中每个节点的两个子树的深度相差不会超过1。
样例
给出二叉树 A={3,9,20,#,#,15,7}
, B={3,#,20,15,7}
A) 3 B) 3
/ \ \
9 20 20
/ \ / \
15 7 15 7
二叉树A是高度平衡的二叉树,但是B不是
解题思路
根据题目需求,左右子树高度差不超过1的是平衡二叉树,一般的做法是,递归求解,但是这样重复计算,我们可以先判断子节点的高度差,如果不是平衡二叉树,那么整棵树不是,否则将会递归求解。
实现代码
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param root: The root of binary tree.
* @return: True if this Binary tree is Balanced, or false.
*/
bool isBalancedHelper(TreeNode *root,int *depth)
{
if(root==NULL)
{
*depth=0;
return true;
}
int left,right;
if(isBalancedHelper(root->left,&left)&&isBalancedHelper(root->right,&right))
{
int diff=left-right;
if(diff<=1&&diff>=-1)
{
*depth=1+(left>right?left:right);
return true;
}
}
return false;
}
bool isBalanced(TreeNode * root) {
// write your code here
int depth;
return isBalancedHelper(root,&depth);
}
};