剑指offer-把二叉树打印成多行

题目描述

• 从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
• 地址： 牛客地址

问题分析

• 这就是层次遍历，只不过要求区分层次
• 那么如何区分层次呢？
  
  • 初始想法是用 null 来区分每一层，作为每一层的结尾标志。遍历完一层便在队列中加一个null，以后该null出队时，判断此时队列是否为空，若为空，则说明已经遍历结束（下一层没有元素），若不为空，则在队列中加null。具体见实现
  • 推荐做法，用一个变量 curLevelSize 来统计当前层的节点数。只要当前已经遍历了 curLevelSize个元素， 便遍历完了一层，然后去更新curLevelSize，遍历下一层。
    那么如何得到 curLevelSize呢？
    一想层序遍历的过程便知道，当将要遍历该层时，此时队列中只有该层的元素，所以值便是当前队列中元素的个数。

经验教训

• 层次遍历如何区分每一层

代码实现

• 用 null 区分每一层

    ArrayList<ArrayList<Integer> > Print1(TreeNode pRoot) {
        ArrayList<ArrayList<Integer> > res = new ArrayList<ArrayList<Integer> >();
        if (pRoot == null) {
            return res;
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        //初始化
        queue.add(pRoot);
        queue.add(null);
        //curList用来存储每一层
        ArrayList<Integer> curList = new ArrayList<>();
        while (! queue.isEmpty()) {
            TreeNode curNode = queue.removeFirst();
            if (curNode == null) {
                if (! queue.isEmpty()) {
                    //未遍历结束才加null，防止无限循环
                    queue.add(null);
                }
                //到null,该层结束
                res.add(curList);
                //新建链表，存储下一层
                curList = new ArrayList<>();
                continue;
            }
            curList.add(curNode.val);
            if (curNode.left != null) {
                queue.add(curNode.left);
            }
            if (curNode.right != null) {
                queue.add(curNode.right);
            }
        }
        return res;
    }

• 用当前层个数来区分每一层(推荐做法)

    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer> > res = new ArrayList<ArrayList<Integer> >();
        if (pRoot == null) {
            return res;
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(pRoot);
        while (! queue.isEmpty()) {
            //得到当前行的大小
            int curLevelSize = queue.size();
            //curLevelList用来存储当前行
            ArrayList<Integer> curLevelList = new ArrayList<>();
            //遍历当前行
            for (int i = 0; i < curLevelSize; i++) {
                TreeNode curPop = queue.removeFirst();
                curLevelList.add(curPop.val);
                if (curPop.left != null) {
                    queue.add(curPop.left);
                }
                if (curPop.right != null) {
                    queue.add(curPop.right);
                }
            }
            res.add(curLevelList);
        }
        return res;
    }