题目:
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
注:和102几乎没有区别,添加一个line变量,根据奇数行还是偶数行,决定正插还是反插。
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> resultList = new ArrayList<>();
if(root == null)
return resultList;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int line = 0;
//将每层的节点放入Queue
while(!queue.isEmpty()) {
LinkedList<Integer> levelList = new LinkedList<>();
//用count来限制每层poll几个节点
int count = queue.size();
for(int i=0 ;i<count;i++){
TreeNode now = queue.poll();
if(line%2==0){
levelList.add(now.val);
}else{
levelList.addFirst(now.val);
}
if(now.left!=null){
queue.add(now.left);
}
if(now.right!=null){
queue.add(now.right);
}
}
resultList.add(levelList);
line++;
}
return resultList;
}