同学去阿里面试,项目问完了,面试官感觉一般,出了下面这道题。往往面试官觉得项目一般,没有太多亮点的时候,就会考察应聘者的基础知识是否扎实,基础知识扎实的话,也能进大厂;一般来讲,项目和基础二者必须有其一,才有进大厂的可能。
这位同学思考了一会儿,顺利的做除了这道题,并且给出了解答,面试官很满意,顺利的拿到了offer。大家好好看下下面这道题。
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如: 给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
提示:
节点总数 <= 1000
解法一:层序遍历 + 双端队列
这道题广度优先搜索,BFS,而众所周知,BFS往往需要借助 队列 的先入先出特性来实现。
/**
* 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>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
if(root != null) queue.add(root);
while(!queue.isEmpty()){
LinkedList<Integer> tmp = new LinkedList<>();
for(int i=queue.size(); i>0; i--){
TreeNode node = queue.poll();
if(res.size() % 2 == 0) tmp.addLast(node.val);//奇数层,顺序添加
else tmp.addFirst(node.val);//偶数层,添加在队列头部
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
}
res.add(tmp);
}
return res;
}
}
解法二:层序遍历 + 倒序
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
if(root != null) queue.add(root);
while(!queue.isEmpty()) {
List<Integer> tmp = new ArrayList<>();
for(int i = queue.size(); i > 0; i--) {
TreeNode node = queue.poll();
tmp.add(node.val);
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
}
if(res.size() % 2 == 1) Collections.reverse(tmp);//倒序
res.add(tmp);
}
return res;
}
}