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

代码随想录

110.平衡二叉树

题解思路

leetcode中以节点为一度，而维基百科以边为一度。
求深度可以从上到下去查：前序遍历，中左右
求高度只能从下到上去查：后续遍历。左右中
根节点的高度就是这棵树的最大深度。
注意求什么就设什么为返回变量，这里要求返回是否是平衡二叉树，也就是高度差是否大于1，那么我们只要返回高度差是否大于1即可。

代码

使用后序遍历可以查高度，通过判断左右子树的高度差来决定是否是平衡的。

class Solution {
    
    public boolean isBalanced(TreeNode root) {
       return getHeight(root)!=-1;
    }

    public int getHeight(TreeNode root){
        if(root == null) return 0;
        
        int leftheight = getHeight(root.left);
        if(leftheight==-1) return -1;
        int rightheight = getHeight(root.right);
        if(rightheight==-1) return -1;

        if(Math.abs(leftheight-rightheight)>1) return -1;

        return Math.max(leftheight,rightheight)+1;

    }
}

257. 二叉树的所有路径

思路

通过前序遍历，获得路径的节点，终止条件是遇到叶子节点
题解思路：
把path暂时存储在一个integer的list中，遇到了叶子节点，再转换成string，放到最终的结果中。
把结果集和临时路径结果都作为递归函数的形参传递 ，以及树的节点也作为形参传递。

代码

class Solution {

  public List<String> binaryTreePaths(TreeNode root) {
        List<String> res = new ArrayList<>();// 存最终的结果
        if (root == null) {
            return res;
        }
        List<Integer> paths = new ArrayList<>();// 作为结果中的路径
        traversal(root, paths, res);
        return res;
    }

    private void traversal(TreeNode root, List<Integer> paths, List<String> res) {
        paths.add(root.val);// 前序遍历，中
        // 遇到叶子结点
        if (root.left == null && root.right == null) {
            // 输出
            StringBuilder sb = new StringBuilder();// StringBuilder用来拼接字符串，速度更快
            for (int i = 0; i < paths.size() - 1; i++) {
                sb.append(paths.get(i)).append("->");
            }
            sb.append(paths.get(paths.size() - 1));// 记录最后一个节点
            res.add(sb.toString());// 收集一个路径
            return;
        }
        // 递归和回溯是同时进行，所以要放在同一个花括号里
        if (root.left != null) { // 左
            traversal(root.left, paths, res);
            paths.remove(paths.size() - 1);// 回溯
        }
        if (root.right != null) { // 右
            traversal(root.right, paths, res);
            paths.remove(paths.size() - 1);// 回溯
        }
    }

}

404.左叶子之和

思路

使用后序遍历，因为，需要求叶子节点的和，所以需要先从叶子节点开始操作
代码思路：递归的函数形参设计为：是否为左节点，以及当前节点。这样就可以在判断是左叶子节点的时候加上值了。

代码

class Solution {
    
    public int sumOfLeftLeaves(TreeNode root) {
       int sum = 0;
       if(root == null) return sum;
       sum = preOrder(root,false);
       return sum;
    }

    public int preOrder(TreeNode cur,Boolean isLeft){
       

       int leftsum =0;
        if(cur.left!=null)
       leftsum  = preOrder(cur.left,true);

        int rightsum=0;
        if(cur.right!=null)
        rightsum = preOrder(cur.right,false);

        int sum= 0;
        if(cur.left == null && cur.right == null){
           if(isLeft){
               sum = cur.val;
               
           }
           
       }
        return leftsum+rightsum+sum;

    }
}