剑指--平衡二叉树

剑指–平衡二叉树

1，题目：

2，思路：

方法一：后序遍历+剪枝操作：

后序遍历+剪枝操作：
此树的深度 等于 左子树的深度 与 右子树的深度 中的 最大值 +1。
dfs：
1.返回值：
2.当节点root 左 / 右子树的深度差 ≤1 ：则返回当前子树的深度，即节点 root 的左 / 右子树的深度最大值 +1 （ max(left, right) + 1 ）；
3.当节点root 左 / 右子树的深度差 >2 ：则返回 -1，代表 此子树不是平衡树 。
4.终止条件：
5.当 root 为空：说明越过叶节点，因此返回高度 0 ；
6.当左（右）子树深度为 -1 ：代表此树的 左（右）子树 不是平衡树，因此剪枝，直接返回 −1 ；

isBalanced(root) 函数：

返回值： 若 dfs(root) != -1 ，则说明此树平衡，返回 true ； 否则返回 false 。

下面是对应的图解：

方法二：先序遍历 + 判断深度 （从顶至底）

    算法流程：

• 1.isBalanced(root) 函数： 判断树 root 是否平衡
• 2.特例处理： 若树根节点 root 为空，则直接返回 true ；
• 3.返回值： 所有子树都需要满足平衡树性质，因此以下三者使用与逻辑 && 连接；
• 4.abs(self.depth(root.left) - self.depth(root.right)) <= 1 ：判断 当前子树 是否是平衡树；
• 5.self.isBalanced(root.left) ： 先序遍历递归，判断 当前子树的左子树 是否是平衡树；
• 6.self.isBalanced(root.right) ： 先序遍历递归，判断 当前子树的右子树 是否是平衡树；
• 7.depth(root) 函数： 计算树 root 的深度
• 8.终止条件： 当 root 为空，即越过叶子节点，则返回高度 0 ；
• 9.返回值： 返回左 / 右子树的深度的最大值 +1 。

下面是对应的图解：

3，代码：

方法一：后序遍历+剪枝操作：

class Solution {
  
    public boolean isBalanced(TreeNode root) {

        /*

        后序遍历+剪枝操作：
        此树的深度 等于 左子树的深度 与 右子树的深度 中的 最大值 +1。
dfs：
        1.返回值：
2.当节点root 左 / 右子树的深度差 ≤1 ：则返回当前子树的深度，即节点 root 的左 / 右子树的深度最大值 +1 （ max(left, right) + 1 ）；
3.当节点root 左 / 右子树的深度差 >2 ：则返回 -1，代表 此子树不是平衡树 。
4.终止条件：
5.当 root 为空：说明越过叶节点，因此返回高度 0 ；
6.当左（右）子树深度为 -1 ：代表此树的 左（右）子树 不是平衡树，因此剪枝，直接返回 −1 ；

isBalanced(root) 函数：

返回值： 若 dfs(root) != -1 ，则说明此树平衡，返回 true ； 否则返回 false 。



        */
        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;
    }
}

方法二：先序遍历 + 判断深度 （从顶至底）

class Solution {
    public boolean isBalanced(TreeNode root) {
        /*

        方法二：先序遍历 + 判断深度 （从顶至底）

        算法流程：
1.isBalanced(root) 函数： 判断树 root 是否平衡

2.特例处理： 若树根节点 root 为空，则直接返回 true ；
3.返回值： 所有子树都需要满足平衡树性质，因此以下三者使用与逻辑 && 连接；
4.abs(self.depth(root.left) - self.depth(root.right)) <= 1 ：判断 当前子树 是否是平衡树；
5.self.isBalanced(root.left) ： 先序遍历递归，判断 当前子树的左子树 是否是平衡树；
6.self.isBalanced(root.right) ： 先序遍历递归，判断 当前子树的右子树 是否是平衡树；
7.depth(root) 函数： 计算树 root 的深度

8.终止条件： 当 root 为空，即越过叶子节点，则返回高度 0 ；
9.返回值： 返回左 / 右子树的深度的最大值 +1 。




        */
        if (root == null) return true;
        return Math.abs(depth(root.left) - depth(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
    }

    private int depth(TreeNode root) {
        if (root == null) return 0;
        return Math.max(depth(root.left), depth(root.right)) + 1;
    }
}