LeetCode-102-二叉树的层序遍历

Arbicoral2023-10-11 13:50

题目描述：

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

解题思路：

使用队列 先进先出的特点存储每次遍历的节点。
代码实现：

java 复制代码

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        bfs(root);
        return res;
    }

	// 层遍历
    private void bfs(TreeNode root){
        if (root==null){
            return;
        }

        // 使用 队列 来遍历并保存，每次记录队列的大小size，
        // 每次弹出一个，在队尾加入该节点的左右孩子，同时size--
        // 当 size=0,说明这一层就结束了，更新size，更新后的 size大小就是下一层节点的个数
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);// 根节点入队列
        while (!queue.isEmpty()) {
            List<Integer> path = new ArrayList<>();// 每更新遍历一层的时候都要新建一个
            int size = queue.size();// 更新 size
            while (size>0){
                // 弹出一个
                TreeNode tmp = queue.poll();
                path.add(tmp.val);// path 记录队列弹出的元素
                // 左孩子入队列
                if (tmp.left != null) {
                    queue.add(tmp.left);
                }
                // 右孩子入队列
                if (tmp.right != null) {
                    queue.add(tmp.right);
                }
                size--;
            }
            // 出while循环说明这一层更新完了
            res.add(path);// 收获结果，不需要新建一个ArrayList<>()，因此每次的都是最新的
        }
    }
}