Binary Tree Zigzag Level Order Traversal 解答

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 public class Solution {
11     public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
12         List<List<Integer>> result = new ArrayList<List<Integer>>();
13         if (root == null)
14             return result;
15         // Set a flag to help judge traversal sequence
16         // flag = 0, from left to right; flag = 1, from right to left
17         int flag = 0;
18         List<TreeNode> current = new ArrayList<TreeNode>();
19         List<TreeNode> next;
20         current.add(root);
21         
22         while (current.size() > 0) {
23             List<Integer> oneRecord = new ArrayList<Integer>();
24             next = new ArrayList<TreeNode>();
25             for (TreeNode tmpNode : current) {
26                 if (tmpNode.left != null)
27                     next.add(tmpNode.left);
28                 if (tmpNode.right != null)
29                     next.add(tmpNode.right);
30                 if (flag == 0)
31                     oneRecord.add(tmpNode.val);
32                 else
33                     oneRecord.add(0, tmpNode.val);
34             }
35             result.add(oneRecord);
36             current = next;
37             flag = 1 - flag;
38         }
39         return result;
40     }
41 }