题目
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof
解题思路
使用一个队列存储每一层的节点,level变量记录第几层,初始level等于1,
每次循环都记录下当前队列的大小(n),n即为每层的节点数量,当队列不为空时,
每次出队n个节点同时保存节点的值到一个list(当level为偶数时使用链表头插法),
每层节点遍历完后level+1,直到队列为空,最后返回结果
代码实现
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res =new ArrayList<>();
if(root == null) return res;
Queue<TreeNode> queue=new LinkedList<>();
queue.offer(root);
int level=1;
while(!queue.isEmpty()){
List<Integer> tmp=new ArrayList<>();
int n=queue.size();
for(int i=0;i<n;i++){
TreeNode node=queue.poll();
if((level & 1)==1){
tmp.add(node.val);
}else{
tmp.add(0,node.val);//头插法
}
if(node.left !=null) queue.offer(node.left);
if(node.right !=null) queue.offer(node.right);
}
level++;
res.add(tmp);
}
return res;
}
}
|