题目:给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7 返回锯齿形层次遍历如下: [ [3], [20,9], [15,7] ]
思路:锯齿状其实就是控制奇数行从左往右扫描,而偶数行就从右往左扫描,则用一个变量标记这种变化,另外用两个 Deqeue 进行保存每一行的 TreeNode,用 List 保存总的数据,最后将其加入 List<List<Integer>> 类型的数据,并将其返回。
代码:
/**
* 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>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> lists = new LinkedList<>();
if(root == null) return lists;
int flag = 0;
Deque<TreeNode> queue = new LinkedList<>();
Deque<TreeNode> nextQueue = new LinkedList<>(); // 保存下一行的 TreeNode
List<Integer> list = new LinkeList<>(); // 保存每一层的 TreeNode
queue.offerLast(root);
while(!queue.isEmpty()){
TreeNode node = queue.pollLast();
list.offerLast(node.val);
if(flag == 0) {
if(node.left != null){
nextQueue.offerLast(node.left);
}
if(node.rigth != null) {
nextQueue.offerLast(node.right);
}
}else{
if(node.rigth != null) {
nextQueue.offerLast(node.right);
}
if(node.left != null){
nextQueue.offerLast(node.left);
}
}
if(queue.isEmpty()){
flag = 1- flag;
Deque<TreeNode> temp = nextQeue;
nextQueue = queue;
queue = temp;
lists.add(list);
list = new LinkedList<>();
}
}
return lists;
}
}