自己写的:
每个结点都递归地去判断深度,时间复杂度应该大于o(n)。
比较好的一点:主函数递归。
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return bool布尔型
*/
int Height(TreeNode* pRoot){
if(pRoot == NULL)
return 0;
return max(Height(pRoot->left),Height(pRoot->right))+1;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
// write code here
if(pRoot == NULL)
return true;
if(abs(Height(pRoot->left)-Height(pRoot->right))<=1 && IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right))
return true;
else
return false;
}
};
在递归计算树的深度的时候,顺便判断一下是否是平衡树。保证树的每个结点遍历一次,满足o(n)。
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return bool布尔型
*/
int flag = 0;
int Judge(TreeNode* pRoot){
if(pRoot == NULL)
return 0;
int hLeft = Judge(pRoot->left);
int hRight = Judge(pRoot->right);
if(abs(hLeft-hRight) > 1)
flag = 1;
return max(hLeft, hRight)+1;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
// write code here
Judge(pRoot);
if(flag == 0)
return true;
else
return false;
}
};
模板的:
思路是我自己写的第一种,感觉时间复杂度没有满足吧???模板不会出错了吧???