给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
每遍历一层节点,就存储该层节点的值,根据顺序标志位判断是在集合尾部添加还是添加到头部,并且存储下一层的左右节点,改变顺序标志位的然后一直遍历结束,直到节点集合为空。
public static List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if (root == null) {
return result;
}
Queue<TreeNode> trees = new LinkedList<>();
trees.add(root);
boolean forward = true;
while (!trees.isEmpty()) {
int size = trees.size();
LinkedList<Integer> list = new LinkedList<>();
for (int i = 0; i < size; i++) {
TreeNode tree = trees.poll();
if (forward) {
list.add(tree.val);
} else {
list.addFirst(tree.val);
}
if (tree.left != null) trees.add(tree.left);
if (tree.right != null) trees.add(tree.right);
}
result.add(list);
forward = !forward;
}
return result;
}