牛客_剑指Offer_编程题 ：把二叉树打印成多行

题目描述

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

解题思路

• 使用一个ArrayList<ArrayList >保存各层次节点
• Queue 暂存下一层节点

循环部分：

• 使用变量levelNum 暂存queue当前层的节点数目
• 每次获取queue队列顶部的节点，从左到右遍历
• 同时使用tmpList保存每一层的数据，添加到resultList中
• 最后返回结果

代码实现

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


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>> resultList = new ArrayList<>();// 存储多层数据
        Queue<TreeNode> queue = new LinkedList<>();    // 暂存下一层节点
        if (pRoot == null) {    
            return resultList;
        }
        int levelNum = 0;
        queue.add(pRoot);    
        while (!queue.isEmpty()){
            levelNum = queue.size();    // 当前层节点个数
            ArrayList<Integer> tmpList = new ArrayList<>();
            while (levelNum > 0){    // 在queue中遍历当前层的所有节点
                TreeNode tmpNode = queue.remove();    // 弹出顶层节点
                if (tmpNode != null){        // 节点不为空，添加其左右节点，如果其左右节点为null，在下一层当中不会添加到queue队列中
                    tmpList.add(tmpNode.val);
                    queue.add(tmpNode.left);
                    queue.add(tmpNode.right);
                }

                levelNum--;
            }
            if (tmpList.size() > 0){
                resultList.add(tmpList);
            }
        }

        return resultList;
    }

}

本文由博客群发一文多发等运营工具平台 OpenWrite 发布