[菜鸟训练]102. 二叉树的层序遍历

最新推荐文章于 2023-11-06 22:00:02 发布

原创最新推荐文章于 2023-11-06 22:00:02 发布 · 150 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

211 篇文章

订阅专栏

Top100

60 篇文章

订阅专栏

本文介绍如何使用广度优先搜索（BFS）算法对给定的二叉树进行层序遍历，提供了一个Java代码示例，并展示了如何返回每个层次的节点值。

题目描述：

给你一个二叉树，请你返回其按层序遍历得到的节点值。（即逐层地，从左到右访问所有节点）。

示例：
二叉树：[3,9,20,null,null,15,7],

返回其层序遍历结果：

[
[3],
[9,20],
[15,7]
]

解题思路：

可以使用广搜来完成该问题。我们先将根节点添加到队列中，然后将其加入到相应的层次列表中，同时将该节点的孩子节点加到队列中，然后重复循环，直到最终遍历完所有点。

代码：

public class LC102 {

    //变型的BFS
    //时间复杂度O(n)   空间复杂度O(n)
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ansList = new ArrayList<>();

        if (root == null){
            return  ansList;
        }

        //使用队列来存储相应节点
        Queue<TreeNode> queue = new LinkedList<>();
        //将根节点加入队列
        queue.add(root);

        while (!queue.isEmpty()){
            //获取每一层的元素
            List<Integer> level = new ArrayList<>();
            //获取当前队列中存在的元素个数
            int size = queue.size();

            //然后将队列的中的元素分别取出来，放到层次列表中，并将它非空的左右孩子入队
            for (int i = 1; i <= size; i++){
                TreeNode node = queue.poll();

                level.add(node.val);
                if (node.left != null){
                    queue.add(node.left);
                }
                if (node.right != null){
                    queue.add(node.right);
                }
            }
            //最后将该层的层次列表存到结果列表中
            ansList.add(level);
        }
        return ansList;
    }
}