Skip to content

Instantly share code, notes, and snippets.

@InterviewBytes
Created June 10, 2017 18:36
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save InterviewBytes/821733a0e83971a45d586bc52d40a752 to your computer and use it in GitHub Desktop.
Save InterviewBytes/821733a0e83971a45d586bc52d40a752 to your computer and use it in GitHub Desktop.
ZigZag level order travsersal
package com.interviewbytes.trees;
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
package com.interviewbytes.trees;
import java.util.*;
public class ZigZagLevelOrder {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<>();
Deque<TreeNode> queue = new LinkedList<>();
if (root != null) queue.offer(root);
boolean reverse = false;
while (!queue.isEmpty()) {
int size = queue.size();
List<Integer> level = new ArrayList<>();
while (size > 0) {
TreeNode node = queue.poll();
level.add(node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
size--;
}
if (reverse) Collections.reverse(level);
list.add(level);
reverse = !reverse;
}
return list;
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment