給定二叉樹,返回 Z 字形層序遍歷:第一層左→右,第二層右→左,交替進行。例:[3,9,20,null,null,15,7] → [[3],[20,9],[15,7]]。
用隊列層序遍歷,每層收集節點值到 ArrayList,若該層為右→左則在列表頭插入(或遍歷完反轉)。代碼較長。
同樣層序遍歷,但根據層號奇偶決定插入位置:
list.add(val)。list.add(0, val)。無需額外棧或反轉,代碼極短。
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) return res;
Queue<TreeNode> q = new LinkedList<>();
q.offer(root);
int level = 1;
while (!q.isEmpty()) {
int size = q.size();
List<Integer> cur = new ArrayList<>();
for (int i = 0; i < size; i++) {
TreeNode node = q.poll();
if (level % 2 == 1) cur.add(node.val);
else cur.add(0, node.val);
if (node.left != null) q.offer(node.left);
if (node.right != null) q.offer(node.right);
}
res.add(cur);
level++;
}
return res;
}
時間 O(n),空間 O(n)。add(0, val) 為 O(k) 但總體仍為 O(n)。