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

题目描述

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

和层次遍历类似，为了把二叉树的每一行单独打印到一行里，需要增加两个变量，一个变量记录当前层中还没有打印的节点数，另一个变量记录下一层的节点数目。

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

/**
 * @author yuan
 * @date 2019/2/14
 * @description 层次遍历，分层打印
 */
public class 把二叉树打印成多行 {
    static class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }
    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        if (pRoot == null) {
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(pRoot);
        // 下一层的节点数
        int nextLevel = 0;
        // 当前层还未打印的节点数
        int toBePrint = 1;

        ArrayList<Integer> curLevel = new ArrayList<>();
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            curLevel.add(node.val);
            // 打印了一个节点
            --toBePrint;

            if (node.left != null) {
                queue.offer(node.left);
                ++nextLevel;
            }
            if (node.right != null) {
                queue.offer(node.right);
                ++nextLevel;
            }

            // 如果当前层全部打印完
            if (toBePrint == 0) {
                result.add(curLevel);
                curLevel = new ArrayList<>();
                // 重新赋值
                toBePrint = nextLevel;
                nextLevel = 0;
            }
        }

        return result;
    }
}