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

题目所属分类

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;

 
    }
}