原题链接
- 层序遍历一个二叉树;就是从左到右一层一层的去遍历二叉树。
- 需要借用一个队列来实现,队列先进先出,符合一层一层遍历的逻辑,而是用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。
- 而这种层序遍历方式就是图论中的广度优先遍历,不过我们应用在二叉树上。
- 实现层序遍历的关键是 每一层都会用一个List来存储,然后将每一层的List,都添加到一个List<List> 中。
- 首先,定义一个队列que,然后将当前的node结点加入到que中。
- 当que非空,定义一个列表itemList,然后记录que的长度len
- 当len大于0时,使用临时结点tmpNode储存que弹出的元素,然后将这个元素的val加入到itemList中,当tmpNode的左右结点都是非空的时候,也加入到que中,然后执行len–
- 如何实现的将每一层都用单独的itemList来保存呢?? 原理如下:
- 当进入第一个while的时候,首先创建了一个itemList,并统计len的长度,比如此时的len长度为1,即只有一个结点,然后进入了第二个while(len > 0),当len一直满足大于0的时候,itemList就会一直记录当前层的结点,len的原始长度就是上次while循环时统计的len,而每次新添加的tmpNode的孩子结点还没有记录进长度len。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root == null) return res;
Queue<TreeNode> que = new LinkedList<>();
que.offer(root);
while(!que.isEmpty()){
int len = que.size();
List<Integer> item = new ArrayList<>();
while(len > 0){
TreeNode node = que.poll();
item.add(node.val);
if(node.left != null) que.offer(node.left);
if(node.right != null) que.offer(node.right);
len--;
}
res.add(item);
}
return res;
}
class Solution {
public List<List<Integer>> resList = new ArrayList<List<Integer>>();
public List<List<Integer>> levelOrder(TreeNode root) {
checkFun02(root);
return resList;
}
public void checkFun02(TreeNode node){
if(node == null) return;
Queue<TreeNode> que = new LinkedList<TreeNode>();
que.offer(node);
while(!que.isEmpty()){
List<Integer> itemList = new ArrayList<>();
int len = que.size();
while(len > 0){
TreeNode tmpNode = que.poll();
itemList.add(tmpNode.val);
if(tmpNode.left != null) que.offer(tmpNode.left);
if(tmpNode.right != null) que.offer(tmpNode.right);
len--;
}
resList.add(itemList);
}
}
}