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

最新推荐文章于 2022-03-09 09:27:47 发布

原创最新推荐文章于 2022-03-09 09:27:47 发布 · 147 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

67 篇文章

订阅专栏

本文介绍了一种二叉树的层次遍历算法，通过使用队列实现从上到下，由左至右的节点输出。该算法适用于任何二叉树结构，并提供了完整的Java代码实现。

题目描述

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

代码

public class Solution60 {


    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {

        ArrayList<ArrayList<Integer>> result = new ArrayList<>();

        if (pRoot==null)
            return result;

        Queue<TreeNode> queue = new LinkedList<>();

        queue.add(pRoot);

        while (!queue.isEmpty()) {

            int start = 0, end = queue.size();

            ArrayList<Integer> layer = new ArrayList<>();

            while (start++ < end) {
                TreeNode cur = queue.poll();
                layer.add(cur.val);
                if (cur.left != null) {
                    queue.add(cur.left);
                }
                if (cur.right != null) {
                    queue.add(cur.right);
                }
            }

            result.add(layer);


        }
        return result;

    }

    public static void main(String[] args) {

        int[] a = {1, 5, 7, 11, 43, 12, 45, 32, 23, 9, 21};
        TreeNode tree = BeanUtil.createCompleteBinaryTree(a);
        BeanUtil.printTreeStructure(tree, tree.val, 0);


        ArrayList<ArrayList<Integer>> res = new Solution60().Print(null);

        for (ArrayList<Integer> arr:res){

            BeanUtil.print(arr);
            BeanUtil.printReturn();


        }

    }


}