二叉树层序遍历(返回一个不定长二维数组)

力扣_二叉树的层序遍历

题目中要求我们返回一个二维数组

List是一个接口,List<List<Integer>>就代表:

在List这个类型的数组中,存储着类型为List<Integer>类型的元素,而List<Integer>又是一个存储着Integer类型的一维数组。

java中支持不定长的二维数组(也就是每一行的元素数量是可以不相同的),因此我们可以直接用List接口,去存储每一层树的元素


好了,现在我们来正式看一下这道题目的解题思路。

先审题:

题目要求层序遍历二叉树,然后把每个节点的元素存储在二维数组中,然后返回这个二维数组。

那改如何实现层序遍历呢?

思路:

可以借助队列,这个数据结构去帮我们层序遍历

定义一个cur,cur先指向根节点,然后建立一个空队列,把cur节点Offer进队列。

下面建立一个循环,只要队列不为空,就一直执行循环

在循环中,用poll方法,出队列一个元素,用cur接收,把cur中的元素存储在二维数组中,至于怎么一层一层的存储最后将,先把整体思路捋一遍,把如何用队列层序遍历讲清楚。

再把cur节点的左节点入队(如果有)和右节点(如果有)分别按左右顺序入队

然后再出队一个元素,用cur接收,我们发现:

按照这个循环走下去,就会呈现出层序遍历。

以上就是用队列实现层序遍历的基本思路。


回到刚才的问题,我们该如果按层序遍历顺序的把每个元素存储在二维数组中?

其实并不难。

先建立一个List<Integer>类型的一维数组,存储每一层的元素。

在刚才演示的循环开头

获取当前队列所存的元素大小,在此循环中,在建立一个循环。

只要size不为零,就出队一个元素,把这个元素存储在List中,同时按照,左右顺序,入队两个元素。

最终,这个循环表达的意思就是,获取的size的大小,就是当前层节点数的大小,只要把size个节点存到一个一维数组中,同时按照左右顺序入队元素即可。

最后再把这个一维数组add到List<List<Integer>>这个二维数组中即可。

完整代码:

复制代码
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {

        List<List<Integer>> ret = new ArrayList<>();
        if (root == null) return ret;//如果树是空的,那么就没必要遍历,返回ret默认的空
        Queue<TreeNode> queue=new LinkedList<>();//创建一个队列用来存储一层的节点

        TreeNode cur=root;
        queue.offer(cur);
        while(!queue.isEmpty()){//队列不为空,就进入循环
            int size=queue.size();//计算队列当前层的大小,然后,分别出队,同时存储在二维数组

            List<Integer> list=new ArrayList();//每一层的数组

            while(size>0){  
                cur=queue.poll();//接收出队的节点,出队一个节点
                list.add(cur.val);//存储在数组中

                if(cur.left!=null){//左子树不是空,入队
                    queue.offer(cur.left);
                }
                if(cur.right!=null){//右子树不为空,入队
                    queue.offer(cur.right);
                }
                size--;
            }
            ret.add(list);
        }
return ret;
    }
}
相关推荐
报错小能手3 分钟前
数据结构 定长顺序表
数据结构·c++
再卷也是菜1 小时前
C++篇(21)图
数据结构·c++·算法
星轨初途1 小时前
C++入门(算法竞赛类)
c++·经验分享·笔记·算法
没书读了2 小时前
数据结构-考前记忆清单
数据结构
灰灰勇闯IT2 小时前
KMP算法在鸿蒙系统中的应用:从字符串匹配到高效系统级开发(附实战代码)
算法·华为·harmonyos
小龙报2 小时前
【算法通关指南:数据结构和算法篇 】队列相关算法题:3.海港
数据结构·c++·算法·贪心算法·创业创新·学习方法·visual studio
csuzhucong2 小时前
一阶魔方、一阶金字塔魔方、一阶五魔方
算法
五花就是菜2 小时前
P12906 [NERC 2020] Guide 题解
算法·深度优先·图论
辞旧 lekkk3 小时前
【c++】封装红黑树实现mymap和myset
c++·学习·算法·萌新
星轨初途3 小时前
C++的输入输出(上)(算法竞赛类)
开发语言·c++·经验分享·笔记·算法