103. Binary Tree Zigzag Level Order Traversal

Posted 2020-07-06 我的名字叫周周

    /*
     * 103. Binary Tree Zigzag Level Order Traversal 
     * 11.21 By Mingyang
     */
        public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
            List<List<Integer>> res = new ArrayList<List<Integer>>();
            if (root == null)
                return res;
            LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
            queue.add(root);
            int num = 0;
            boolean reverse = false;// a flag
            while (!queue.isEmpty()) {
                num = queue.size();
                List<Integer> levelres = new ArrayList<Integer>();
                for (int i = 0; i < num; i++) {
                    TreeNode node = queue.poll();
                    levelres.add(node.val);
                    if (node.left != null)
                        queue.add(node.left);
                    if (node.right != null)
                        queue.add(node.right);
                }
                if (reverse) {
                    Collections.reverse(levelres);
                    reverse = false;
                } else {
                    reverse = true;
                }
                res.add(levelres);
            }
            return res;
        }