代码随想录算法训练营Day17

110. 平衡二叉树

class Solution {

    public boolean isBalanced(TreeNode root) {
        // 如果获取根节点高度的结果不为-1，则表示平衡，否则不平衡
        return getHeight(root) != -1;
    }

    private int getHeight(TreeNode root) {
        // 如果当前节点为空，高度为 0
        if (root == null) {
            return 0;
        }
        // 获取左子树高度
        int leftHeight = getHeight(root.left);
        // 如果左子树高度计算结果为 -1（即左子树不平衡），则直接返回 -1
        if (leftHeight == -1) {
            return -1;
        }
        // 获取右子树高度
        int rightHeight = getHeight(root.right);
        // 如果右子树高度计算结果为 -1（即右子树不平衡），则直接返回 -1
        if (rightHeight == -1) {
            return -1;
        }
        // 如果左右子树高度差大于 1，说明不平衡，返回 -1
        if (Math.abs(leftHeight - rightHeight) > 1) {
            return -1;
        }
        // 返回当前节点的高度，为左右子树高度最大值加 1
        return Math.max(leftHeight, rightHeight) + 1;
    }
}

257. 二叉树的所有路径

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) {
        if (root == null) { // 如果根节点为空，直接返回 0
            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;
    }
}