题目
分析
判断二叉树是否为平衡树的关键就是每个节点的左右节点的树高差不超过1,也即对一个节点,其| leftHeight-rightHeight |<=1且其左节点和右节点也满足上述情况,以此类推。清楚了这一点后,代码就很容易写出来了。判断条件为若你存在有某节点的左右树高差大于1的,那么便不是二叉树;若你遍历了所有节点,都符合不大于1的条件,那么就是二叉树。而说到遍历,联系上述内容,我们就很容易想到DFS的思想。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
bool isBalanced(struct TreeNode* root) {
return nodeHeight(root)!=-1?true:false;
}
int nodeHeight(struct TreeNode* root){
int leftHeight,rightHeight;
if(root==NULL)
return 0;
if(root->left==NULL)
leftHeight=0;
else
leftHeight=nodeHeight(root->left);
if(root->right==NULL)
rightHeight=0;
else
rightHeight=nodeHeight(root->right);
if(abs(leftHeight-rightHeight)<=1 && leftHeight!=-1 && rightHeight!=-1)
return (leftHeight>rightHeight?leftHeight:rightHeight)+1;
else
return -1;
}