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

题目描述：

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

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

3
/ \
9  20
  /  \
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;
    }
}