Leetcode-每日一题【剑指 Offer 32 - III. 从上到下打印二叉树 III】

题目

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

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

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [20,9],
  [15,7]
]

提示：

1. 节点总数 <= 1000

解题思路

1.题目要求我们实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。此题与【剑指 Offer 32 - II. 从上到下打印二叉树 II】的解题思想几乎一致，大家可以先去学习一下。

2.唯一的区别就是此题打印的顺序不太一样，我们经过分析可以发现，奇数行的元素是正着打印的，偶数行的元素是倒着打印的。所以我们需要设置一个变量 sum 来记录我们的行数，在往临时动态数组 cur 中插入元素时，就需要判断一下这个sum是奇数还是偶数，若为奇数，我们就直接add插入，若为偶数，我们就使用addFirst进行头插，这样插入的顺序就是反着的。要注意的一点是，我们需要将临时数组 cur 变为LinkedList ，因为只有LinkedList才有头插法。

代码实现

class Solution {
    public List> levelOrder(TreeNode root) {
        if(root == null){
            return new ArrayList<>();
        }

        Queue queue = new LinkedList<>();
        List> res = new ArrayList<>();
        int sum = 1;
        queue.add(root);
        while(!queue.isEmpty()){
            int k = queue.size();
            LinkedList cur = new LinkedList<>();
            for(int i = 0; i < k; i++ ){
                TreeNode t = queue.poll();
                if(sum % 2 == 1){
                    cur.add(t.val);
                }else{
                    cur.addFirst(t.val);
                }
                if(t.left != null) queue.add(t.left);
                if(t.right != null) queue.add(t.right);
            } 
            sum ++;
            res.add(cur);
        }
        return res;

    }
}

测试结果