剑指Offer-面试题32-III-从上到下打印二叉树III

2020-05-01

题目描述

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

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

返回其层次遍历结果：

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

提示：节点总数 <= 1000

思路

同II一样，借助队列对二叉树进行层序遍历，因为需要之字形打印，因此每层得到的一维数组，需要判断当前层是奇数层还是偶数层，如果是偶数层，则一维数组需要倒序。根据result.size()可判断上一层是奇数层还是偶数层，以此得出当前层是奇数层还是偶数层。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 class Solution{
     public List<List<Integer>> levelOrder(TreeNode root){
         Queue<TreeNode> queue = new LinkedList<>();
         List<List<Integer>> list = new ArrayList<>();
         if(root != null){
             queue.add(root);
         }
         while(!queue.isEmpty()){
             List<Integer> temp = new ArrayList<>();
             for(int i=queue.size();i>0;i--){
                 TreeNode node = queue.poll();
                 temp.add(node.val);
                 if(node.left != null){
                     queue.add(node.left);
                 }
                 if(node.right != null){
                     queue.add(node.right);
                 }
             }
             //判断上一层是奇数层还是偶数层，如果是奇数层，则当前层为偶数层，需要倒序后再加入result中
             if(result.size() % 2 == 1){
                 Collections.reverse(temp);
             }
             result.add(temp);
         }
         return result;
     }
 }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。