给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下:[
[3],
[20,9],
[15,7]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
【层次遍历】
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
if (root == null) return ans;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
int size = 1;
List<Integer> res = new ArrayList<>();
res.add(root.val);
ans.add(res);
boolean reverse = true;
while (!queue.isEmpty()) {
int curSize = 0;
List<Integer> res1 = new ArrayList<>();
while (size-- > 0) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
res1.add(node.left.val);
curSize++;
}
if (node.right != null) {
queue.offer(node.right);
res1.add(node.right.val);
curSize++;
}
}
if (res1.isEmpty()){
break;
}
if (reverse) {
Collections.reverse(res1);
reverse = false;
}else{
reverse = true;
}
ans.add(res1);
size = curSize;
}
return ans;
}