zl程序教程

您现在的位置是:首页 >  数据库

当前栏目

102. 二叉树的层序遍历

2023-04-18 16:11:54 时间

思路:

每次把下一层的数据从左到右放进去,然后取的时候容器里的就是当前层的数据,先把大小拿到,然后取数据同时将非空左右孩子放进去,直到取完该层的数据

代码:

class Solution {
        public List<List<Integer>> levelOrder(TreeNode root) {
            List<List<Integer>> res = new ArrayList<>();
            if (root == null) {
                return res;
            }
            Queue<TreeNode> queue = new LinkedList<>();
            queue.offer(root);

            while (!queue.isEmpty()) {
                List<Integer> temp = new ArrayList<>();

                int levelSize = queue.size();
                //每次将当前层的结点从左向右取出,将下一层的结点从左向右放进去
                for (int i = 0; i < levelSize; i++) {
                    TreeNode curr = queue.poll();
                    temp.add(curr.val);

                    if (curr.left != null) {
                        queue.offer(curr.left);
                    }
                    if (curr.right != null) {
                        queue.offer(curr.right);
                    }
                }
                res.add(temp);
            }

            return res;
        }
    }