55.平衡二叉树_节点的深度相差不超过1-CSDN博客

本文链接：https://blog.csdn.net/lushylin/article/details/117021141

题目：

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

示例：

解题思路：

这个题需要自底向上做遍历，可以直接用递归来做。但是一边要记录树的高度一边又要记录是否满足二叉树条件。这个时候千万不要想着一个函数用来传深度int一个函数用来传是否满足二叉得boolean。最好用int做返回，true的时候返回深度，false的时候返回-1即可。
自底向上的方式可以选择后序遍历:左子节点-右子节点-根节点
也可以选择：右子节点-左子节点-根节点(这个在树的右视图里会考)。
所以递归里面就按照先返回本节点的深度为目的，如果出现递归传上来是-1那么直接返回-1，如果传递上来的左右子树高度差不超过1则返回最高子树深度+1即可。

参考：

Krahets55.平衡二叉树
Java代码

//下面代码是摘自Krahets本题解下的评论：(ID:莫非)
//因为ta很细心的加了标注，我就摘过来参考了
public class Solution55_2 {

    public boolean isBalanced(TreeNode root) {
        return dfs(root)==-1?false:true;
    }

    //用left，right记录root左右子节点的深度，避免遍历root时对左右节点的深度进行重复计算。
    //考虑到需要同时记录各个节点的深度和其是否符合平衡性要求，这里的返回值设为int,用一个特殊值-1来表示出现不平衡的节点的情况，而不是一般采用的boolean
    public int dfs(TreeNode root){
    	//用后序遍历的方式遍历二叉树的每个节点（从底至顶）,先左子树，再右子树，最后根节点，
        if(root==null) return 0;//root等于0时，该节点符合要求，返回其深度0，而不返回-1；
        int left = dfs(root.left);//left最开始的取值为0，从底朝上遍历，先左子树，后右子树，最后根节点
        if(left==-1) return -1;//若出现节点的深度为-1，则进行剪枝，开始向上返回，之后的迭代不再进行
        int right = dfs(root.right);
        if(right==-1) return -1;
        return Math.abs(right-left)<2?Math.max(left,right)+1:-1;//+1不能少
        //最开始计算的是左子树最左侧的一个叶节点，其左右子节点不存在，left=0，right=0，满足条件，返回该叶节点的深度max(0,0)+1=1;
    }
}