Sword32-III——从上到下打印二叉树 III
方法1——BFS
- 思路:此题也称为之字形遍历,与上一题的解法类似。只是需要判断当前是奇数偶数层,奇数层则tmp可不变,偶数层则tmp需逆序(或者使用双端队列从头部弹出),此处采取双端队列tmp
- 如何判断是奇数层还是偶数层?
- 通过res的size进行判断,当res的size大小为偶数,说明正在遍历的就是奇数层;反之为奇数,说明正在遍历的为偶数层
- 特殊情况与临界分析:根节点为空
- 终止条件:队列中不存在元素
- 步骤:
- 定义一个存储结果res的ArrayList,便于校验中直接返回空ArrayList
- 根节点为空校验
- 定义一个辅助队列
- 定义一个移动节点
- 将移动节点(即根节点)入队
- 循环条件:队列非空
- while循环
- 定义一个**双端队列(LinkedList充当)**用于存储每一层结果
- for循环,因为queue的size存入元素后将会变化,因此需要记录当前queue的size大小,因此将i直接记录queue.size(),使得i依次减1即可,完成之后即弹出上一层的所有节点
- 从队列中弹出一个元素
- 判断res的size大小是奇数还是偶数
- 结果为偶数,说明当前为奇数层,将其加入到tmp的尾部
- 结果为奇数,说明当前为偶数层,将其加入到tmp的首部
- 判断该节点左子节点是否非空,非空则入队
- 判断该节点右子节点是否非空,非空则入队
- 此时tmp已经装满了上一层的结果,将其加入到res中
- 返回结果(此处不需要转换为数组形式)
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
TreeNode cur = root;
queue.offer(cur);
while (!queue.isEmpty()) {
LinkedList<Integer> tmp = new LinkedList<>();
for (int i = queue.size(); i > 0; i--) {
cur = queue.poll();
if ((res.size() & 1) == 0) {
tmp.addLast(cur.val);
} else {
tmp.addFirst(cur.val);
}
if (cur.left != null) {
queue.offer(cur.left);
}
if (cur.right != null) {
queue.offer(cur.right);
}
}
res.add(tmp);
}
return res;
}
![方法1——BFS](https://i-blog.csdnimg.cn/blog_migrate/2eab9e95531fdea4c42c7fea3205eb1a.png)