题目
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
提示:
节点总数 <= 1000
解题思路
1.题目要求我们实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。此题与【剑指 Offer 32 - II. 从上到下打印二叉树 II】的解题思想几乎一致,大家可以先去学习一下。
2.唯一的区别就是此题打印的顺序不太一样,我们经过分析可以发现,奇数行的元素是正着打印的,偶数行的元素是倒着打印的。所以我们需要设置一个变量 sum 来记录我们的行数,在往临时动态数组 cur 中插入元素时,就需要判断一下这个sum是奇数还是偶数,若为奇数,我们就直接add插入,若为偶数,我们就使用addFirst进行头插,这样插入的顺序就是反着的。要注意的一点是,我们需要将临时数组 cur 变为LinkedList ,因为只有LinkedList才有头插法。
代码实现
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null){
return new ArrayList<>();
}
Queue<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
int sum = 1;
queue.add(root);
while(!queue.isEmpty()){
int k = queue.size();
LinkedList<Integer> cur = new LinkedList<>();
for(int i = 0; i < k; i++ ){
TreeNode t = queue.poll();
if(sum % 2 == 1){
cur.add(t.val);
}else{
cur.addFirst(t.val);
}
if(t.left != null) queue.add(t.left);
if(t.right != null) queue.add(t.right);
}
sum ++;
res.add(cur);
}
return res;
}
}
测试结果