【算法】力扣 剑指 Offer 32. II. 从上到下打印二叉树 II

---

题目

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

---

思路

主要考察的还是层次遍历二叉树的算法，但是我觉得难点是在需要按层输出上面。
看了解析，觉得在 while 循环里嵌套一个 for 循环非常妙，一开始也有想到for循环，可是却卡在了不知道如何判断层数上。

---

代码实现

while 循环里嵌套的 for 循环，在一次 while 循环结束后的辅助队列中的 size() 就是当前层数的元素个数，但是如果是用从低到高的 for 循环写法，辅助队列的 size() 会变，并且也不再表示当前层数的元素个数了。所以采用从高到低的 for 循环写法，提前定死边界。

public List<List<Integer>> levelOrder(TreeNode root) {
        //创建层次遍历存储List
        List<List<Integer>> levelOrderTree = new LinkedList<>();
        //创建辅助队列
        Queue<TreeNode> bfsTree = new LinkedList<>();
        //健壮性检验
        if(root != null){
            bfsTree.offer(root);
        }
        
        while(!bfsTree.isEmpty()){
            //创建每一层的存储List
            List<Integer> bfsList = new LinkedList<>();
            //层次遍历 for循环从高到低
            for(int i = bfsTree.size();i > 0;i--){
                root = bfsTree.poll(); //获得根节点
                bfsList.add(root.val); //添加进当前层的遍历List中
                if(root.left != null){ //添加左孩子
                    bfsTree.offer(root.left);
                }
                if(root.right != null){ //添加右孩子
                    bfsTree.offer(root.right);
                }
            }
            levelOrderTree.add(bfsList); //结束一层遍历
        }

        return levelOrderTree;
    }

---

总结

采用从高到低的 for 循环写法，提前定死边界。