力扣:平衡二叉树java
主要问题:
采用后序遍历。要先获取左右子树的高度,判断是否为平衡二叉树,
与求最大深度不同。在每次获取左右节点的深度之后,立刻判断返回值是否为-1,也就是这颗子树是否为平衡树,不是则直接返回-1,不用进行下面的判断。
流程:
递归三部曲:
选择返回值和参数(返回高度或者-1,参数为根节点)
结束条件:节点为空返回0,空节点的高度为0
单层递归逻辑:获取左子树高度,判断高度为-1,则说明不是平衡树,直接返回-1,判断右子树高度,判断高度为-1,则说明不是平衡树,直接返回-1,比较左右子树高度差值为-1到1之间,返回最大值加一,否则说明不是平衡树,返回-1。
代码:
class Solution {
public boolean isBalanced(TreeNode root) {
if(isB(root) == -1) return false;//返回值为-1,说明不是平衡二叉树
else{
return true;//返回值为高度,则说明是平衡二叉树
}
}
public int isB(TreeNode root){//递归函数,返回值为高度,
//高度为-1则不是平衡二叉树
if(root == null) return 0;//结束条件,空节点返回0高度
int lH = isB(root.left);//获取左孩子节点
if(lH == -1) return -1;//判断是否左孩子节点高度是否为-1,直接返回,说明不是平衡二叉树
int rH = isB(root.right);//获取右孩子
if(rH == -1) return -1;//判断是否右孩子节点高度是否为-1,直接返回,说明不是平衡二叉树
if((lH-rH)>=-1&&(lH-rH)<=1){//左右子树高度在-1到1之间则返回最大高度+1
return 1+Math.max(lH,rH);
}else{//否则说明不是平衡二叉树
return -1;
}
}
}