本文共 1598 字，大约阅读时间需要 5 分钟。

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

分析

前面的文章提到，在二叉树的层序遍历中，使用Queue队列的“先进先出”特性保存访问节点的左右子节点，只需要使用一个ArrayList保存遍历的结果输出即可。

此处与前面的二叉树层序遍历类似但又不同的是，每一行需要单独打印。那么每一行都是一个单独的ArrayList存储，将每一个单独的ArrayList添加到总ArrayList中即可。此外还需要两个变脸：一个变量表示当前层中还没有打印的节点数，另一个变量表示下一层的节点数。比较总ArrayList的size和二叉树的层数可以判断是否需要产生新的单独ArrayList（新行）。

牛客AC：

import java.util.ArrayList;import java.util.Queue;import java.util.LinkedList;/*public class TreeNode {    int val = 0;    TreeNode left = null;    TreeNode right = null;    public TreeNode(int val) {        this.val = val;    }}*/public class Solution {
       ArrayList
   
    
      > Print(TreeNode pRoot) {        ArrayList
     
      
       > list = new ArrayList
       
        
         >(); if(pRoot == null) return list; int countLevel = 0; // 二叉树的层数 int countNodeToBePrinted = 1; // 当前层尚未打印的节点数 int countNodeNextLevel = 0; // 下一层的节点数 // 队列记录节点的左右子树节点 // LinkedList实现了Queue和Stack接口 Queue
         
           queue = new LinkedList
          
           (); queue.offer(pRoot); while(!queue.isEmpty()) { // list的size小于层数，需要增加新的list，保存下一层 if(list.size() <= countLevel) list.add(new ArrayList
           
            ()); ArrayList
            
              curList = list.get(countLevel); TreeNode pNode = queue.poll(); curList.add(pNode.val); // 相当于打印，则尚未打印的节点数减去1 countNodeToBePrinted--; // 左右子节点不为空，则入队 if(pNode.left != null) { queue.offer(pNode.left); countNodeNextLevel++; } if(pNode.right != null) { queue.offer(pNode.right); countNodeNextLevel++; } // 当前层全部打印完毕，转到下一层，更新变量 if(countNodeToBePrinted == 0) { countLevel++; countNodeToBePrinted = countNodeNextLevel; countNodeNextLevel = 0; } } return list; }}
            
           
          
         
        
       
      
     
    
   

参考

转载地址：http://tkxgi.baihongyu.com/