LC 102. 二叉树的层序遍历

题干

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

示例 1：

输入： root = [3,9,20,null,null,15,7]
输出： [[3],[9,20],[15,7]]

思路

主要考察点：二叉树的层序遍历

• 怎么区分不同层的节点？ 第N层二叉树的节点数总和 = 队列的长度
• 使用队列来辅助完成遍历

解题

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null){
            return new ArrayList<>();
        }
        List<List<Integer>> res = new ArrayList<>();
        // 使用队列来完成层序遍历
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(!queue.isEmpty()){
            int num = queue.size();
            List<Integer> list = new ArrayList<>();
            while(num>0){
                // 出队列，将它下一层的子树添加到队列中
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left!=null){
                    queue.add(node.left);
                }
                if(node.right!=null){
                    queue.add(node.right);
                }
                // 通过这个计数来区别不同层节点
                num--;
            }
            // 遍历完二叉树的一层，将结果添加到结果集
            res.add(list);
        }
        return res;
    }
}