代码随想录算法训练营day17|平衡二叉树、二叉树的所有路径、左叶子之和

2023.3.31 平衡二叉树

110. 平衡二叉树 - 力扣（LeetCode）

要判断一个二叉树是否为平衡二叉树，我们首先要判断其根节点的左右两个子树的高度差不能超过1，

再去分别判断左右两个子节点左右子树高度差不超过1，以此类推。

因此我们可以使用递归的方式来完成。首先判断返回类型和传入参数。

显然，返回类型是布尔类型用于判断左右两个子树是否满足平衡二叉树

传入参数，需要传入左子节点和右子节点分别作为左右子树的根节点？

终止条件，当左子节点和右子节点同时为空，返回true，

当左子节点和右子节点都不为空，继续进行判断。

当有一个节点为空，继续判断，传入另一个不为空节点的参数

调用计算高度的方法来判断是否为平衡二叉树。

第一版错误代码：

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null){
            return true;
        }
        return Math.abs(getDepth(root.left)-getDepth(root.right)) <2?true:false;
    }

    //获取节点高度的方法
    public int getDepth(TreeNode root){
        if(root == null){
            return 0;
        }
        int leftDepth = getDepth(root.left);
        int rightDepth = getDepth(root.right);
        return Math.max(leftDepth,rightDepth)+1;
    }
}

仅仅考虑了根节点左右子树的情况，有可能存在左右两个子树高度符合，但是左右子树里面存在高度不符合的情况。因此当判断完左右两个子树以后，我们需要继续进行判断。

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null){
            return true;
        }
        if(Math.abs(getDepth(root.left)-getDepth(root.right))>=2){
            return false;
        }
        //继续判断剩余子节点
        return isBalanced(root.left) && isBalanced(root.right);

    }

    //获取节点高度的方法
    public int getDepth(TreeNode root){
        if(root == null){
            return 0;
        }
        int leftDepth = getDepth(root.left);
        int rightDepth = getDepth(root.right);
        return Math.max(leftDepth,rightDepth)+1;
    }
}

但是上面的方法会导致时间复杂度有点高，因为我们在获取高度的方法里面遍历了一次二叉树，在判断是否平衡里面又遍历了一次二叉树，导致对整个二叉树结构遍历了两次。

那么能否只遍历一次就能得到结果呢？我们可以采用类似后续遍历的方法，同样的得到节点高度，当左子节点的高度和右子节点的高度差不满足要求时，返回负一，那么只有存在不满足条件的值的时候，最后的根节点返回值一定为-1.

class Solution {
    public boolean isBalanced(TreeNode root) {
      return height(root) >= 0;
    }
    //获取节点高度的方法
    public int height(TreeNode root){
        if(root == null){
            return 0;
        }
        int left = height(root.left);
        int right = height(root.right);
        if(left == -1 || right == -1 || Math.abs(left-right) >1){//不满足条件的几种情况
            return -1;
        }
        return Math.max(left,right)+1;
    }
}

2023.3.31 二叉树的所有路径

257. 二叉树的所有路径 - 力扣（LeetCode）

后序遍历出路径，记录进StringBuffer中，然后反转？但是我们需要记录很多条路径，怎样吧不同路径记录进不同的StringBuffer中？

利用前序遍历的思路，每遍历到一个节点，判断是否为叶子结点，如果是，直接加入结果集，如果不是，继续遍历后续节点。

显然，我们需要传入当前头结点，以及一个String用来记录路径，同时还要传入res结果集来接收结果。

class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> res = new ArrayList<>();
        //开始传入一个空字符串作为路径
        getPath(root,"",res);
        return res;
    }
    public void getPath(TreeNode root,String path,List<String> res){
        if(root == null){
            return;
        }
        //首先要将当前节点的值计入路径中,同时传入上次记录下来的路径放入我们的StringBuffer以便增删
        StringBuffer onePath = new StringBuffer(path);
        onePath.append(Integer.toString(root.val));
        //接下来判断节点的情况
        if(root.left ==null && root.right == null){//当前节点为叶节点
            //结束判断，将路径放入结果集
            res.add(onePath.toString());
        }
        else{//当前节点不是叶子结点，则继续记录左右子节点的路径
            //路径与路径之前使用->连接
            onePath.append("->");
            //这里我们需要传入上次记录下来的路径onePath，继续当做一个StringBuffer来接收后续节点
            getPath(root.left,onePath.toString(),res);
            getPath(root.right,onePath.toString(),res);
        }
    }
}

2023.3.31 左叶子之和

404. 左叶子之和 - 力扣（LeetCode）

怎样才能确定是左叶子节点?

左边的子节点，要想获得整个二叉树的左叶子节点之和，我们需要分别获得左右子树的左子叶节点之和。

class Solution {
    int sum = 0;
    public int sumOfLeftLeaves(TreeNode root) {
        if(root == null){
            return 0;
        }
        //判断左叶子节点
        if(root.left != null && root.left.left == null && root.left.right == null){
             sum += root.left.val;
        }
         //遍历左子树和右子树
         sumOfLeftLeaves(root.left);
         sumOfLeftLeaves(root.right);
         return sum;
    }
}

借用前序遍历的思路完成即可。重点在于确定左子叶节点的逻辑。