[牛客101] 二叉树的层序遍历
![[牛客101] 二叉树的层序遍历](http://pic.ttrar.cn/nice/%5b%e7%89%9b%e5%ae%a2101%5d%e4%ba%8c%e5%8f%89%e6%a0%91%e7%9a%84.jpg)
这道题会考察很多知识点,这里专门进行详解
文章目录
- 题目描述
- 二. 题目分析
- 完整代码
题目描述
![[牛客101] 二叉树的层序遍历](https://img-blog.csdnimg.cn/95875eed7b024fe1b10fbce8974e695c.png)
![[牛客101] 二叉树的层序遍历](https://img-blog.csdnimg.cn/d1673ef4f748431686570c3124697021.png)
二. 题目分析
首先,我们会想到存储方式为二维数组.数组每一行存储一层的结点.怎么确定每一行要存储几个结点呢.由于节点与节点之间存在父子关系,所以,在存储某一层的结点时,就可以通过存储这一层节点的左右孩子,来记录下一层的结点.
首先,定义二维数组,数组每一行存储一层的结点.
ArrayList<ArrayList<Integer>> list = new ArrayList<>();
先把一层的结点先放到队列中,记录好一层节点的数目size
![[牛客101] 二叉树的层序遍历](https://img-blog.csdnimg.cn/b914ef38019c4771a48391de1eee03d8.png)
再依次取出来,放到ArrayList里,用size控制好这一层的数目.由于队列先进先出,不会打乱节点顺序.
在取出结点的同时,将结点的左右孩子结点放入队列,一层的节点全部放进去之后,会再次记录size值.如下图
![[牛客101] 二叉树的层序遍历](https://img-blog.csdnimg.cn/728a5d83f45245acbcf2ecdcb7561fba.png)
之后,重复操作,取出这一层的节点值,同时,把他们的左右孩子放进队列,进行计数.
![[牛客101] 二叉树的层序遍历](https://img-blog.csdnimg.cn/da8a2054986e48d29178dacc8da18e71.png)
直到队列为空,结束循环.
![[牛客101] 二叉树的层序遍历](https://img-blog.csdnimg.cn/4f0d827a9c9548f0a9258e2e17487842.png)
完整代码
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {// 层序遍历ArrayList<ArrayList<Integer>> list = new ArrayList<>();Queue<TreeNode> queue = new LinkedList<>();if(root == null){return list;}//把根节点放进去queue.offer(root);while(!queue.isEmpty()){int size = queue.size();//注意arr的位置,一定要放在外循环里ArrayList<Integer> arr = new ArrayList<>();while(size > 0){TreeNode cur = queue.poll();size--;if(cur != null){arr.add(cur.val);}if(cur.left != null){queue.offer(cur.left);}if(cur.right != null){queue.offer(cur.right);}}list.add(arr);}return list;}


