leetcode 102. 二叉树的层序遍历-java实现

最新推荐文章于 2024-08-31 09:55:36 发布

依嘫_吃代码

最新推荐文章于 2024-08-31 09:55:36 发布

阅读量529

点赞数

分类专栏： LeetCode 文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/qq_41810415/article/details/127479039

版权

LeetCode 专栏收录该内容

158 篇文章 4 订阅

订阅专栏

这篇博客详细介绍了二叉树的层序遍历（BFS）的Java实现模板，强调了在循环中使用队列时，应当注意队列大小（len=q.size()）的提前获取，以确保正确输出层序遍历结果。此外，提供了具体的代码示例，展示了如何遍历二叉树并构建层序遍历的输出。

摘要由CSDN通过智能技术生成

题目所属分类

BFS层序遍历的模板

原题链接

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

代码案例：
输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

题解

简单的BFS模板但是要注意的是 len = q.size() 如果没有的话 for循环里面直接为q.size（）会导致结果有出入 q的范围会随着下面的if左右节点还有的话会增加的

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        if(root == null) return ans;
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.add(root);
        while(!q.isEmpty())
        {
            List<Integer> res = new ArrayList<Integer>();
            int len = q.size();
            //这里面的q.size() 一定要先标出来  否则出来的结果不符合
            for(int i = 0;i < len;i ++)
            {
                TreeNode t = q.poll();
                res.add(t.val);
                if(t.left != null) q.add(t.left);
                if(t.right != null) q.add(t.right);
            }
            ans.add(res);
        }   
        return ans;

 
    }
}