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

简介: 代码随想录算法训练营第十四天 | LeetCode 102. 二叉树的层序遍历、LeetCode 226. 翻转二叉树、LeetCode 101. 对称二叉树

1. LeetCode 102. 二叉树的层序遍历

1.1 思路

  1. 二叉树的层序遍历就相当于图论里的广度优先搜索,之前的递归遍历就相当于图论里的深度优先搜索
  2. 只依赖二叉树的结构本身是无法做到层序遍历的,因此需要借助一个队列的数据结构
  3. 首先将根节点放入,每一层要记录当时队列的长度,这个长度就相当于这层有几个元素,然后根据这个长度把每一层的元素弹出放入一个集合中,因为层序遍历返回的是List<List<Integer>>,就相当于一个二维数组,每一层就是每一维,如果没有这个长度记录那么队列既有这一层又有下一层时就混乱了
  4. 在弹出元素时再判断这个元素是否有左右孩子,有就放入队列,在原本层里的元素都弹出后再记录新的长度,每弹出一个元素长度就--

1.2 代码

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

2. LeetCode 226. 翻转二叉树

2.1 思路

  1. 应该用什么遍历方式?递归还是非递归?递归的话是前中后序?要把这些想清楚
  2. 前序和后序是最方便的,中序也可以但很麻烦
  3. 确定递归的参数和返回值:返回值TreeNode,参数TreeNode 根节点
  4. 终止条件:遇到空节点了,if(root==null)return root;
  5. 处理逻辑:以前序遍历为例,前序遍历是“根左右”,那么交换就是swap(root.left,root.right),然后递归调用函数,分别是invertTree(root.left),invertTree(root.right)
  6. 如果是后序遍历,那么就把swap写在左右子树的下面即可
  7. 为什么中序不可以呢?其实如果是把swap写在中间的话,那么先把左子树翻转了,再把左右子树swap,那么此时的左子树就是原来的右子树,但接下来是invertTree(root.right),就相当于又处理原来的左子树了,即原来的右子树就没有处理,所以我们把invertTree(root.right)改为invertTree(root.left)即可

2.2 代码

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;
    }
}

3. LeetCode 101. 对称二叉树

3.1 思路

  1. 二叉树类的题目确定遍历顺序是很重要的,这题我们用递归的方式,应该用前中后哪个呢?这题只能用后序,后序是“左右根”,因为我们要不断收集左右孩子的信息返回给上一个(即父节点)节点,再收集左右孩子的信息返回给上一个节点,这样才能知道以这个左节点为根节点的树和右节点为根节点的树是否是可以相互翻转的。因为只有后续遍历,才能知道把底部的孩子的信息是否相等的信息返回给上一层
  2. 确认递归的参数和返回值:由于上述所说,是否是对称二叉树就是判断左右子树是否是可以翻转的,因此定义一个函数,传入左子树和右子树,返回值就是是否是可以翻转的
  3. 终止条件:如果左节点为空右节点不为空,就是不可翻转的就return false;如果左节点不为空右节点为空,同理return false;如果左右都为空就是可翻转的return true;如果左右都不为空但值不相等,就return false;如果左右节点不为空且值相等就继续向下遍历
  4. 处理单层递归的逻辑:因为我们要比较外侧节点和内侧节点,如果相同就为true。于是递归调用函数boolean outside= compare(left.left , right.right),这就是左孩子的左孩子和右孩子的右孩子,即外侧节点的比较;然后再比较内侧节点,boolean inside=compare(left.right , right.left),这就是左孩子的右孩子和右孩子的左孩子,即内侧节点
  5. 最后是用布尔值result=outside&&inside,同时外侧和内侧相同,这才说明是对称的,最后return result

3.2 代码

class Solution {
    public boolean isSymmetric(TreeNode root) {
        return compare(root.left,root.right);
    }
    public boolean compare(TreeNode left,TreeNode right){
        if(left==null&&right!=null)return false;
        else if(left!=null&&right==null)return false;
        else if(left==null&&right==null)return true;
        else if(left.val!=right.val)return false;
        boolean outside=compare(left.left,right.right);
        boolean inside=compare(left.right,right.left);
        boolean result=outside&&inside;
        return result;
    }
}
相关文章
|
6天前
leetcode代码记录(二叉树的所有路径
leetcode代码记录(二叉树的所有路径
12 0
|
6天前
leetcode代码记录(对称二叉树 中序遍历+回文串 为什么不行
leetcode代码记录(对称二叉树 中序遍历+回文串 为什么不行
8 0
|
6天前
leetcode代码记录(二叉树的最小深度
leetcode代码记录(二叉树的最小深度
10 0
|
6天前
leetcode代码记录(二叉树的最大深度
leetcode代码记录(二叉树的最大深度
9 0
|
6天前
leetcode代码记录(翻转二叉树
leetcode代码记录(翻转二叉树
8 0
|
6天前
leetcode代码记录(二叉树的层序遍历
leetcode代码记录(二叉树的层序遍历
12 0
|
6天前
|
算法
leetcode代码记录(二叉树递归遍历
leetcode代码记录(二叉树递归遍历
8 0
|
6天前
|
算法 数据安全/隐私保护 计算机视觉
基于二维CS-SCHT变换和LABS方法的水印嵌入和提取算法matlab仿真
该内容包括一个算法的运行展示和详细步骤,使用了MATLAB2022a。算法涉及水印嵌入和提取,利用LAB色彩空间可能用于隐藏水印。水印通过二维CS-SCHT变换、低频系数处理和特定解码策略来提取。代码段展示了水印置乱、图像处理(如噪声、旋转、剪切等攻击)以及水印的逆置乱和提取过程。最后,计算并保存了比特率,用于评估水印的稳健性。
|
2天前
|
算法
m基于BP译码算法的LDPC编译码matlab误码率仿真,对比不同的码长
MATLAB 2022a仿真实现了LDPC码的性能分析,展示了不同码长对纠错能力的影响。短码长LDPC码收敛快但纠错能力有限,长码长则提供更强纠错能力但易陷入局部最优。核心代码通过循环进行误码率仿真,根据EsN0计算误比特率,并保存不同码长(12-768)的结果数据。
20 9
m基于BP译码算法的LDPC编译码matlab误码率仿真,对比不同的码长
|
4天前
|
算法
MATLAB|【免费】融合正余弦和柯西变异的麻雀优化算法SCSSA-CNN-BiLSTM双向长短期记忆网络预测模型
这段内容介绍了一个使用改进的麻雀搜索算法优化CNN-BiLSTM模型进行多输入单输出预测的程序。程序通过融合正余弦和柯西变异提升算法性能,主要优化学习率、正则化参数及BiLSTM的隐层神经元数量。它利用一段简单的风速数据进行演示,对比了改进算法与粒子群、灰狼算法的优化效果。代码包括数据导入、预处理和模型构建部分,并展示了优化前后的效果。建议使用高版本MATLAB运行。