题解 | #判断是不是平衡二叉树#
判断是不是平衡二叉树
https://www.nowcoder.com/practice/8b3b95850edb4115918ecebdf1b4d222
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
#include <cstddef>
#include <ios>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return bool布尔型
*
*
**/
int height(TreeNode * root ){
if(root == NULL){
return 0;
}
return max(height(root->left),height(root->right)) + 1;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
// write code here
if(!pRoot){
return true;
}
int left_height = height(pRoot->left);
int right_height = height(pRoot->right);
if(left_height - right_height < -1 || left_height - right_height > 1){
return false;
}
return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
}
};
首先通过递归计算出根节点左子树的高度和根节点右子树的高度,判断根节点的左子树和右子树高度差是否小于1或者大于-1.
然后判断根节点的左子树和根节点的右子树是否是平衡二叉树即可。
拼多多集团-PDD公司福利 817人发布
查看16道真题和解析