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

2020-05-01

题目描述

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

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

返回其层次遍历结果：

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

提示：节点总数 <= 1000

思路

同I一样，借助队列对二叉树进行层序遍历，因为需要分层打印，所以每一层需要按这层的节点数来存储进一个一维数组，一层打印结束后，需要将该一维数组放入二维数组result中。

代码

/**
 * 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>> result = 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){
                     temp.add(node.left);
                 }
                 if(node.right != null){
                     temp.add(node.right);
                 }
             }
             result.add(temp);
         }
         return result;
     }
 }

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