剑指offer 把二叉树打印成多行

最新推荐文章于 2020-03-28 23:41:58 发布

原创最新推荐文章于 2020-03-28 23:41:58 发布 · 178 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#把二叉树打印成多行 #剑指Offer

剑指Offer 专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种二叉树的层次遍历算法，通过使用队列实现从上到下，由左至右的层次遍历，并将每层节点值分别存储在不同的列表中，便于后续处理或展示。

题目描述

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

和层次遍历类似，为了把二叉树的每一行单独打印到一行里，需要增加两个变量，一个变量记录当前层中还没有打印的节点数，另一个变量记录下一层的节点数目。


import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

/**
 * @author yuan
 * @date 2019/2/14
 * @description 层次遍历，分层打印
 */
public class 把二叉树打印成多行 {
    static class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }
    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        if (pRoot == null) {
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(pRoot);
        // 下一层的节点数
        int nextLevel = 0;
        // 当前层还未打印的节点数
        int toBePrint = 1;

        ArrayList<Integer> curLevel = new ArrayList<>();
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            curLevel.add(node.val);
            // 打印了一个节点
            --toBePrint;

            if (node.left != null) {
                queue.offer(node.left);
                ++nextLevel;
            }
            if (node.right != null) {
                queue.offer(node.right);
                ++nextLevel;
            }

            // 如果当前层全部打印完
            if (toBePrint == 0) {
                result.add(curLevel);
                curLevel = new ArrayList<>();
                // 重新赋值
                toBePrint = nextLevel;
                nextLevel = 0;
            }
        }

        return result;
    }
}