代码随想录算法训练营第十五天|层序遍历、226.翻转二叉树、101. 对称二叉树

层序遍历

题目链接：102.二叉树的层序遍历

思路：定义队列，当根节点不为空，将第一个节点加入队列。当队列不为空，首先记录当前层节点个数。定义新的数组，将一层节点放入该数组。遍历当前层元素，将队列弹出元素后存入数组，size--。随后将节点左右孩子加入队列。将数组加入定义的二维数组中。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        if(root == null)return new ArrayList<>();
        List<List<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(!queue.isEmpty()){
            int count = queue.size();
            List<Integer> list = new ArrayList<Integer>();
            while(count > 0){
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left != null)
                    queue.add(node.left);
                if(node.right != null)
                    queue.add(node.right);
                count--;
            }
            res.add(list);
        }
        return res;
    }
}

226.翻转二叉树

思路：判断节点是否为null，交换节点的左孩子和右孩子，分别循环调用原函数向左向右遍历。（递归法）

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if(root == null){
            return null;
        }
        swap(root);
        invertTree(root.left);
        invertTree(root.right);
        return root;
    }

    public void swap(TreeNode root){
        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;
    }
}

101. 对称二叉树

思路：判断二叉树的左子树和右子树是否可以翻转。左节点和右节点都为空 return true；左节点为空或右节点为空 return false；左节点的值不等于右节点的值 return false。向下一层遍历比较外层与内层是否相同，左节点的左孩子和右节点的右孩子比较；左节点的右孩子和右节点的左孩子比较。（递归法）

class Solution {
    public boolean isSymmetric(TreeNode root) {
        if(root==null)return true;
        return match(root.left,root.right);
    }
    public boolean match(TreeNode left,TreeNode right){
        if(left==null&&right==null){
            return true;
        }else if(left==null||right==null){
            return false;
        }
        if(left.val!=right.val){
            return false;
        }
        boolean matchOutside = match(left.left, right.right);
        boolean matchInside = match(left.right, right.left);
        return matchOutside && matchInside;
    }
}