[剑指offer刷题] AcWing 45. 之字形打印二叉树(模拟,栈的应用)
作者:互联网
思路
分层层次遍历
在每一层结束后,向队列中加入一个null
之字型打印
在每次转换层时,对level中的结果反转
题目
请实现一个函数按照之字形顺序从上向下打印二叉树。
即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
数据范围
树中节点的数量 [0,1000]
。
样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, 4, null, null, null, null]
8
/
12 2
/
6 4
输出:[[8], [2, 12], [6, 4]]
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> printFromTopToBottom(TreeNode root) {
boolean flag = false;
List<List<Integer>> res = new ArrayList<List<Integer>>();
if (root == null) return res;
Queue<TreeNode> q = new LinkedList<TreeNode>();
q.offer(root);
q.offer(null);
List<Integer> level = new ArrayList<Integer>();
while (q.size() > 0) {
TreeNode t = q.poll();
if (t == null)
{
if (level.size() == 0) break;
if (flag) Collections.reverse(level);
res.add(new ArrayList<Integer>(level));
level.clear();
q.offer(null);
// 取反操作必须在 #26 行之后
flag = !flag; // 取反操作
continue;
}
level.add(t.val);
if (t.left != null)
q.offer(t.left);
if (t.right != null)
q.offer(t.right);
}
return res;
}
}
标签:之字形,TreeNode,level,res,offer,打印,二叉树,null 来源: https://blog.csdn.net/qq_22473333/article/details/122012758