平衡二叉树
建立概念(维基百科):
二叉树结点的深度:指从根结点到该结点的最长简单路径边的条数
二叉树结点的高度:指从该结点到叶子结点的最长简单路径边的条数
力扣中的概念是路径中所包含的结点数,如下(图为代码随想录中讲解所用图):
求深度,可以从上到下去查,所以需要使用前序遍历(中左右),而高度只能从下到上去查,所以只能后序遍历(左右中),在求二叉树的最大深度中,使用了后序遍历,那是因为根结点的高度就是二叉树的最大深度
题解:
递归:后序遍历
求高度,由上面分析可知,应该用后序遍历
递归三部曲:
1、明确递归函数的参数和返回值
参数为传入的结点指针,返回值为根节点树的高度,返回-1时,说明当前传入结点为根结点的二叉树已经不是平衡二叉树了。
2、明确终止条件
遇到空结点时终止,返回0,表示当前结点为根结点的树高度为0
3、明确单层递归的逻辑
求得左右子树的高度差绝对值是否小于等于1,若是,则返回高度,若不是,则返回-1,表明已经不是平衡二叉树了
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int getHight(TreeNode* node){
//空结点直接返回0
if(node == NULL){
return 0;
}
//求得左右子树的高度
int leftH = getHight(node->left);
int rightH = getHight(node->right);
//判断是否已经不是平衡二叉树了
if(leftH == -1 | rightH == -1){
return -1;
}
//求高度差
if(abs(leftH-rightH)<=1){
return 1+max(leftH,rightH);
}
//否则 不是平衡二叉树了 返回-1
return -1;
}
bool isBalanced(TreeNode* root) {
//空结点直接返回true
if(root == NULL){
return true;
}
//判断是否是平衡二叉树 -1则不是
if(getHight(root) == -1){
return false;
}
//否则 是
return true;
}
};
不能用到层序遍历了,层序遍历只能用于求深度,不能用于求高度