算法的学习笔记—把二叉树打印成多行(牛客JZ78)

举报
尘觉 发表于 2024/08/21 21:27:32 2024/08/21
【摘要】 在算法面试中,二叉树的层序遍历是一个经典的题目。而这道题的要求是进一步将二叉树的每一层结点值打印成多行,即同一层结点从左至右输出,最终结果存放到一个二维数组中返回。接下来,我们将通过代码实例详细解析解题思路。

😀前言
在算法面试中,二叉树的层序遍历是一个经典的题目。而这道题的要求是进一步将二叉树的每一层结点值打印成多行,即同一层结点从左至右输出,最终结果存放到一个二维数组中返回。接下来,我们将通过代码实例详细解析解题思路。

😊把二叉树打印成多行

NowCoder

🥰题目描述

给定一个节点数为 n 的二叉树,要求从上到下按层打印二叉树的 val 值,同一层节点从左至右输出,每一层输出一行,将输出的结果存放到一个二维数组中返回。

以二叉树 {1,2,3,#,#,4,5} 为例,它的层序遍历结果如下:

[
 [1],
 [2,3],
 [4,5]
]

image.png

数据范围

  • 二叉树的节点数 0 ≤ n ≤ 1000
  • 每个节点的值 0 ≤ val ≤ 1000

要求

  • 时间复杂度:O(n)
  • 空间复杂度:O(n)

😋示例

示例1

输入:{1,2,3,#,#,4,5}

返回值:[[1],[2,3],[4,5]]

示例2

输入:{8,6,10,5,7,9,11}

返回值:[[8],[6,10],[5,7,9,11]]

示例3

输入:{1,2,3,4,5}

返回值:[[1],[2,3],[4,5]]

示例4

输入:{}

返回值:[]

🤔解题思路

要实现这一功能,我们可以借助广度优先搜索(BFS)来逐层遍历二叉树。BFS 通常使用队列来实现。具体步骤如下:

  1. 队列初始化:首先,将二叉树的根节点放入队列。
  2. 逐层遍历:循环遍历队列,每次从队列中取出当前层的所有节点,记录它们的值,并将其左右子节点依次加入队列,以便下一次循环遍历。
  3. 存储每层节点:在遍历每层节点时,将当前层的节点值存入一个列表,然后将这个列表添加到最终的结果二维数组中。
  4. 处理空树的情况:如果输入的树为空,则返回一个空的结果数组。

😉代码实现

下面是使用 Java 实现的完整代码:

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

public class Solution {
    public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        // 初始化一个二维数组来保存最终结果
        ArrayList<ArrayList<Integer>> ret = new ArrayList<>();
        // 初始化一个队列用于层序遍历
        Queue<TreeNode> queue = new LinkedList<>();
        // 将根节点加入队列
        queue.add(pRoot);
        
        // 当队列不为空时,继续进行遍历
        while (!queue.isEmpty()) {
            // 用于存储当前层的节点值
            ArrayList<Integer> list = new ArrayList<>();
            // 当前层的节点数量
            int cnt = queue.size();
            // 遍历当前层的所有节点
            while (cnt-- > 0) {
                // 取出队列中的节点
                TreeNode node = queue.poll();
                // 如果节点为空,则跳过
                if (node == null)
                    continue;
                // 将节点的值加入当前层的列表
                list.add(node.val);
                // 将节点的左子节点加入队列
                queue.add(node.left);
                // 将节点的右子节点加入队列
                queue.add(node.right);
            }
            // 如果当前层有节点,将其加入最终结果
            if (list.size() != 0)
                ret.add(list);
        }
        // 返回最终结果
        return ret;
    }
}

详细解析

  1. Queue 数据结构:我们使用 Queue<TreeNode> 来保存每一层的节点。在每一层开始时,队列中包含了所有该层的节点。
  2. 层序遍历:通过 queue.size() 获取当前层的节点数量,并使用一个循环来遍历这些节点。在遍历过程中,将当前节点的左右子节点依次加入队列。
  3. 结果存储:每一层遍历完后,我们将当前层的节点值列表 list 添加到结果数组 ret 中。
  4. 边界条件:如果 queue 中没有元素,即队列为空时,表示树已经遍历完毕,退出循环。

😄总结

这个解法的核心是利用队列来实现二叉树的层序遍历。通过逐层记录节点值,我们能够按照题目要求将每一层节点的值按行输出,并最终返回一个包含多行的二维数组。该解法具有良好的时间和空间复杂度,适用于题目要求的节点数量范围。

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。