代码随想录刷题day17｜平衡二叉树&二叉树的所有路径&左叶子之和

---

day17学习内容

day17主要内容

• 平衡二叉树
• 二叉树的所有路径
• 左叶子之和

---

声明
本文思路和文字，引用自《代码随想录》

一、平衡二叉树

104.原题链接

1.1、思路

• 判断是否平衡二叉树，只要判断左右子树深度是否大于1。大于1就不是平衡二叉树了。

1.2、错误写法

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);
        int rightHeight = getHeight(root.right);
        if (Math.abs(rightHeight - leftHeight) > 1) {
            return -1;
        }
        return Math.max(leftHeight, rightHeight) + 1;
    }
}

1.3、正确写法

后序遍历

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);
        int rightHeight = getHeight(root.right);

        if (leftHeight == -1) {
            return -1;
        }
        if (rightHeight == -1) {
            return -1;
        }

        if (Math.abs(rightHeight - leftHeight) > 1) {
            return -1;
        }
        return Math.max(leftHeight, rightHeight) + 1;
    }
}

二、二叉树的所有路径

257.原题链接

2.1、思路

• 本题需要回溯法
• 收集路径的过程，遍历到叶子结点就可以结束。

2.2、正确写法

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();
            for (int i = 0; i < paths.size(); i++) {
                sb.append(paths.get(i));
                if (i < paths.size() - 1) sb.append("->");
            }
            // 将当前路径添加到结果集中
            res.add(sb.toString()); 
        }

        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); // 删除当前节点，实现回溯
        }
    }
}

2.2.1 测试用例

public class Huisu {

    /**
     * 主方法，用于触发递归遍历二叉树并找到所有从根节点到叶子节点的路径。
     */
    public static 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 static void traversal(TreeNode root, List<Integer> paths, List<String> res) {
        paths.add(root.val); // 将当前节点加入路径中
        System.out.println("当前遍历节点: " + root.val); // 打印当前遍历的节点
        System.out.println("当前路径: " + paths); // 打印当前路径的状态

        if (root.left == null && root.right == null) { // 如果是叶子节点
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < paths.size(); i++) {
                sb.append(paths.get(i));
                if (i < paths.size() - 1) sb.append("->");
            }
            res.add(sb.toString()); // 将当前路径添加到结果集中
            System.out.println("当前结果集: " + res); // 打印当前结果集
        }

        if (root.left != null) {
            traversal(root.left, paths, res); // 递归遍历左子树
            System.out.println("删除前(回溯前)路径: " + paths); // 打印回溯前的路径
            paths.remove(paths.size() - 1); // 删除当前节点，实现回溯
            System.out.println("回溯到节点: " + (paths.isEmpty() ? "根节点" : paths.get(paths.size() - 1))); // 打印回溯到的节点
        }

        if (root.right != null) {
            traversal(root.right, paths, res); // 递归遍历右子树
            System.out.println("删除前(回溯前)路径: " + paths); // 打印回溯前的路径
            paths.remove(paths.size() - 1); // 删除当前节点，实现回溯
            System.out.println("回溯到节点: " + (paths.isEmpty() ? "根节点" : paths.get(paths.size() - 1))); // 打印回溯到的节点
        }
    }

    // 二叉树的节点定义
    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    public static void main(String[] args) {
        TreeNode node7 = new TreeNode(7, null, null);
        TreeNode node6 = new TreeNode(15, null, null);
        TreeNode node3 = new TreeNode(20, node6, node7);
        TreeNode node2 = new TreeNode(9, null, null);
        TreeNode node1 = new TreeNode(3, node2, node3);
        System.out.println(binaryTreePaths(node1));
    }
}

三、左叶子之和

404.原题链接

3.1 什么是左叶子

• 叶子结点就是没有左右孩子节点的节点。
• 左叶子：一定是父节点的左孩子。

3.3.2 思路

• 一定是叶子结点
• 一定是父节点的左节点，
• 满足以上2个条件才能叫左叶子
• 通过递归地遍历二叉树的每个节点，识别并累加所有符合条件的左叶子节点的值，从而计算出所有左叶子节点的总和

3.3.3 代码

略

总结

1.感想

• 二叉树的所有路径看的懵懵的。。。

2.思维导图

本文思路引用自代码随想录，感谢代码随想录作者。