代码随想录算法训练营Day15

102. 二叉树的层序遍历

class Solution {
    // 定义用于存储结果的列表
    public List<List<Integer>> resList = new ArrayList<List<Integer>>();

    // 层次遍历的方法
    public List<List<Integer>> levelOrder(TreeNode root) {
        checkFun(root,0);
        return resList;
    }

    // DFS（深度优先搜索）--递归方式
    public void checkFun(TreeNode node, Integer deep) {
        // 如果节点为空，直接返回
        if (node == null) return;
        // 深度加 1
        deep++;

        // 如果结果列表的大小小于当前深度
        if (resList.size() < deep) {
            // 创建一个新的列表项并添加到结果列表中，利用列表索引表示层级
            List<Integer> item = new ArrayList<Integer>();
            resList.add(item);
        }
        // 将当前节点的值添加到对应深度的列表中
        resList.get(deep - 1).add(node.val);

        // 递归遍历左子树
        checkFun(node.left, deep);
        // 递归遍历右子树
        checkFun(node.right, deep);
    }
}

226. 翻转二叉树

class Solution {
    /**
     * 前后序遍历都可以
     * 中序不行，因为先左孩子交换孩子，再根交换孩子（做完后，右孩子已经变成了原来的左孩子），再右孩子交换孩子（此时其实是对原来的左孩子做交换）
     */
    public TreeNode invertTree(TreeNode root) {
        // 如果根节点为空，直接返回空
        if (root == null) {
            return null;
        }
        // 递归处理左子树
        invertTree(root.left);
        // 递归处理右子树
        invertTree(root.right);
        // 交换当前节点的左右子节点
        swapChildren(root);
        // 返回处理后的根节点
        return root;
    }

    private void swapChildren(TreeNode root) {
        // 定义临时节点，用于交换时暂存左子节点
        TreeNode tmp = root.left;
        // 将左子节点设置为右子节点
        root.left = root.right;
        // 将右子节点设置为暂存的左子节点
        root.right = tmp;
    }
}

101. 对称二叉树

class Solution {
    // 判断二叉树是否对称
    public boolean isSymmetric(TreeNode root) {
        // 调用比较函数，比较根节点的左右子树
        return compare(root.left, root.right);
    }

    // 比较两个子树是否对称
    private boolean compare(TreeNode left, TreeNode right) {
        // 比较为空的情况 
        if (left == null && right!= null) {
            return false;
        }
        if (left!= null && right == null) {
            return false;
        }
        if (left == null && right == null) {
            return true;
        }
        // 不为空，比较值
        if (left.val!= right.val) {
            return false;
        }
        // 比较外侧，即将左子树的左孩子和右子树的右孩子进行对比
        boolean compareOutside = compare(left.left, right.right);
        // 比较内侧，即将左子树的右孩子和右子树的左孩子进行对比
        boolean compareInside = compare(left.right, right.left);
        // 返回比较结果
        return compareOutside && compareInside;
    }
}