class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
boolean flag = true;
while (!queue.isEmpty()) {
int size = queue.size();
List<Integer> levelList = new ArrayList<>();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if (flag) {
levelList.add(node.val);
} else {
levelList.add(0, node.val);
}
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
res.add(levelList);
flag = !flag;
}
return res;
}
}
在这个实现中,我们使用了一个队列 queue 来存储当前层的节点,使用了一个变量 flag 来记录当前层数的遍历顺序,使用了一个变量 levelList 来存储当前层的节点值。如果 flag 为 true,则从左往右遍历当前层,将节点值添加到 levelList 的末尾;如果 flag 为 false,则从右往左遍历当前层,将节点值添加到 levelList 的开头。每次遍历完一层后,将 flag 取反,以便下一层遍历时反转遍历顺序。希望这个实现对您有所帮助,如果您还有其他问题,请随时向我提问。