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

题目链接

思路：

在二叉树层次遍历的基础上进行修改，每次弹出这一整层的节点，得到每一层的所有节点

先将根节点加入队列中，进入循环时，每一层的节点个数等于当前队列的大小，利用此性质来得到当前层的所有节点

    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<>();
        if (root == null) {
            return ans;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            // 当前层的大小，即节点个数，等于当前队列的大小
            int size = queue.size();
            List<Integer> curLevel = new ArrayList<>();
            while (size > 0) {
                TreeNode cur = queue.poll();
                curLevel.add(cur.val);
                // 有左孩子
                if (cur.left != null) {
                    queue.add(cur.left);
                }
                // 有右孩子
                if (cur.right != null) {
                    queue.add(cur.right);
                }
                size--;
            }
            // 当前层结果加入最终结果
            ans.add(curLevel);
        }
        return ans;
    }
    class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }