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

一、平衡二叉树

力扣

思路：求高度使用后续遍历，求深度则使用前序遍历，所以先写左右子树的递归部分并且判断左右子树是不是平衡二叉树，在递归部分之后写逻辑部分即左右子树高度差大于1返回-1，在主函数里判断最后返回的值是否是-1，是的话则不是平衡二叉树。

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

    private 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;
        }
        // 左右子树高度差大于1，return -1表示已经不是平衡树了
        if (Math.abs(leftHeight - rightHeight) > 1) {
            return -1;
        }
        return Math.max(leftHeight, rightHeight) + 1;
    }
}

二、二叉树的所有路径

力扣

思路：一个回溯的题目，先定义好全局path集合和result集合，终止条件是遍历到叶子结点，将path添加进resulr中，因为要输出从上到下的路径，所以使用前序遍历，递归部分则是递归访问左右子树，由于防止终止条件出现空指针错误，所以在递归前判断节点是否为空，然后便是回溯部分，直接从path中移除最后一个元素。

class Solution {
    List<Integer> path = new ArrayList<>();
    List<String> result = new ArrayList<>();
    public List<String> binaryTreePaths(TreeNode root) {
        backTracing(root, path, result);
        return result;
    }

    private void backTracing(TreeNode root, List<Integer> path, List<String> result) {
        path.add(root.val);
        //终止条件，遍历到叶子节点了
        if(root.left == null && root.right == null) {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < path.size() - 1; i++) {
                sb.append(path.get(i)).append("->");
            }
            sb.append(path.get(path.size() - 1));
            result.add(sb.toString());
            return;
        } 
        //怕上面的if判断报空指针异常，所以不能让空节点进入递归
        if(root.left != null) {
            backTracing(root.left, path, result);
            path.remove(path.size() - 1);// 回溯
        }
        if(root.right != null) {
            backTracing(root.right, path, result);
            path.remove(path.size() - 1);// 回溯
        }
    }
}

三、左叶子之和

力扣

思路：采用后序遍历，递归三部曲，返回条件为遇见空节点往上返回，按照后序，先进行左右递归，接受递归返回的值，逻辑处理部分为访问左叶子，也就是root的左孩子是一个叶子结点，将此叶子节点的值赋给midValue，再return总的sum也就是各个左叶子的和。

class Solution {
    public int sumOfLeftLeaves(TreeNode root) {
        if (root == null) return 0;
        int leftValue = sumOfLeftLeaves(root.left);    // 左
        int rightValue = sumOfLeftLeaves(root.right);  // 右
                                                       
        int midValue = 0;
        if (root.left != null && root.left.left == null && root.left.right == null) { 
            midValue = root.left.val;
        }
        int sum = midValue + leftValue + rightValue;  // 中
        return sum;
    }
}