LeetCode刷题——102. 二叉树的层序遍历

热身题——二叉树的层序遍历

题目描述：

题目链接：https://leetcode.cn/problems/binary-tree-level-order-traversal/description/

题目思路：

①第一层循环控制节点加入队列
②第二层循环，遍历每一层的节点
size非常关键，size记录了每一层节点的数量，从而控制了第二层循环中的层序遍历

代码和注释：

/**
 * 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<List<Integer>> res = new LinkedList<>();
        //使用队列来存节点
        Queue<TreeNode> q = new LinkedList<>();
        //让根节点入队
        if(root != null){
            q.offer(root);
        }
        //第一层循环，遍历根节点
        while(q.size() != 0){
            //获取队列本层有几个节点
            int size = q.size();
            //一层的结果
            List<Integer> a = new LinkedList<>();
            //遍历本层的元素
            while(size > 0){
                //获取队首元素，并且让其出队
                TreeNode node = q.poll();
                //将元素值加入到本层结果集中
                a.add(node.val);
                //如果有左右孩子，让其左右孩子入队
                if(node.left != null){
                    q.offer(node.left);
                }
                if(node.right != null){
                    q.offer(node.right);
                }

                //本层元素--
                size--;
            }
            //将一层元素加入到二维的结果集中
            res.add(a);
        }

        //返回最终的结果
        return res;

    }
}

运行结果

刷题时间：2024.03.06