题解 | 判断是不是平衡二叉树:自底向上后序遍历剪枝
判断是不是平衡二叉树
https://www.nowcoder.com/practice/8b3b95850edb4115918ecebdf1b4d222
思路:
设置一个辅助函数,平衡时直接返回子树的高度,不平衡时返回-1,主函数检查返回值是否为-1
步骤:
- 如果为空,返回值为0;
- 递归计算左子树的高度(int left=checkHeight(root->left)),如果左子树返回值为-1,直接返回-1;
- 递归计算右子树,同理左子树
- 计算左右子树的高度差,如果大于1,也return -1;
知识点:
求整型数据的绝对值:abs() 方法在C语言中,只对int整型生效,作用是
正确答案:
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
#include <valarray>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return bool布尔型
*/
int checkHeight(TreeNode* root){
if(root==nullptr) return 0;
int left=checkHeight(root->left);
if(left==-1) return -1;
int right=checkHeight(root->right);
if(right==-1) return -1;
//abs() 方法在C语言中,只对int整型生效,作用是求整型数据的绝对值。
if(abs(left-right)>1) return -1;
return max(left,right)+1;//height
}
bool IsBalanced_Solution(TreeNode* pRoot) {
// write code here
return checkHeight(pRoot)!=-1;
}
};