题解五十三

面试题32 - III. 从上到下打印二叉树 III

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

（来源：力扣（LeetCode））

思路：
题目的要求是从上往下打印，奇数行正序、偶数行倒序打印。使用层序遍历和队列。前面代码依旧实现的是层序遍历，只需要额外实现一个转置功能。
我们设置reverse函数用来转置数组和boolean变量来判断当前层要不要转置。flag每次打印一层值就会取反，决定要不要转置数组（true不转置，false转置）， 第一层不需要转置，初始化为true。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        if(root == null) {
            return new ArrayList<>(0);
        }
        List<List<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
       
        boolean flag = true;               
        while(!queue.isEmpty()){
            List<Integer> list = new ArrayList<>();
            for(int i = queue.size(); i > 0; i --){
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left != null) {
                    queue.add(node.left);
                }
                if(node.right != null) {
                     queue.add(node.right);
                }
            }
          
            if(!flag) {
                 list = reverse(list);
            }
            // 每遍历一层将flag取反，实现一层转置，下一层不转置；
            flag = !flag;
            res.add(list);
        }
        return res;
    }
    
    public List<Integer> reverse(List<Integer> list){
        List<Integer> nums = new ArrayList<>();
        for(int i = list.size() - 1; i >= 0; i --){
            nums.add(list.get(i));                    
        }
        return nums;
    }
}