剑指offer（63）：多行打印二叉树

最新推荐文章于 2022-05-08 21:15:53 发布

smile4lee

最新推荐文章于 2022-05-08 21:15:53 发布

阅读量536

点赞数

分类专栏：剑指offer 文章标签：剑指offer 二叉树层序遍历多行打印

本文链接：https://blog.csdn.net/u011080472/article/details/51290716

版权

剑指offer 专栏收录该内容

70 篇文章 0 订阅

订阅专栏

题目描述

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

分析

前面的文章剑指offer（24）：从上往下打印二叉树提到，在二叉树的层序遍历中，使用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<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null)
            return list;

        int countLevel = 0;     // 二叉树的层数
        int countNodeToBePrinted = 1;   // 当前层尚未打印的节点数
        int countNodeNextLevel = 0;     // 下一层的节点数

        // 队列记录节点的左右子树节点
        // LinkedList实现了Queue和Stack接口
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(pRoot);
        while(!queue.isEmpty()) {
            // list的size小于层数，需要增加新的list，保存下一层
            if(list.size() <= countLevel)
                list.add(new ArrayList<Integer>());

            ArrayList<Integer> 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;
    }

}