在数组中输出二叉树（Java）（leetcode 655）

题面：

在一个 m*n 的二维字符串数组中输出二叉树，并遵守以下规则：

行数 m 应当等于给定二叉树的高度。
列数 n 应当总是奇数。
根节点的值（以字符串格式给出）应当放在可放置的第一行正中间。根节点所在的行与列会将剩余空间划分为两部分（左下部分和右下部分）。你应该将左子树输出在左下部分，右子树输出在右下部分。左下和右下部分应当有相同的大小。即使一个子树为空而另一个非空，你不需要为空的子树输出任何东西，但仍需要为另一个子树留出足够的空间。然而，如果两个子树都为空则不需要为它们留出任何空间。
每个未使用的空间应包含一个空的字符串""。
使用相同的规则输出子树。

输入:
     1
    / \
   2   3
    \
     4
输出:
[["", "", "", "1", "", "", ""],
 ["", "2", "", "", "", "3", ""],
 ["", "", "4", "", "", "", ""]]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/print-binary-tree
 

 

public static List<List<String>> l655(TreeNode root) {
        int height = getHeight(root);
        int weight = (int) (Math.pow(2, height) - 1);
        List<List<String>> result = new ArrayList<>(height);
        //声明一个队列储存节点
        LinkedList<TreeNode> linkedList = new LinkedList<>();
        linkedList.add(root);
        //层数
        int ceng = 1;
        //节点数
        int count = 0;
        //声明能存一个层的数据字符串数组；
        String[] cenl = new String[weight];

        //树的每一层的第一个节点（包括空节点）的位置
        int start = (int)Math.pow(2, height - ceng) - 1;
        //树的某一层节点间（包括空节点）的距离
        int mod = start * 2 + 2;
        while (ceng <= height) {
            //取出队列首
            TreeNode head = linkedList.poll();
            //节点数++
            count++;

            //如果不是空节点
            if (head != null) {
                //将其左右孩子入队（包括空孩子）
                linkedList.add(head.left);
                linkedList.add(head.right);
                String val = String.valueOf(head.val);
                //将节点值填入相应位置
                cenl[start] = val;
            } else {
                //即使该节点是空的，也要将他的两个空孩子入队，占位
                linkedList.add(null);
                linkedList.add(null);
            }
            //确定下一个数组中应该填入的位置
            start = start + mod;
            //判断是否该跳入下一层，重置变量
            if (Math.pow(2, ceng) - 1 == count) {
                //层数++
                ceng++;
                //补全没有值的位置
                for (int i = 0; i < weight; i++) {
                    if (cenl[i] == null) {
                        cenl[i] = "";
                    }
                }
                result.add(Arrays.asList(cenl));
                cenl = new String[weight];
                start = (int)Math.pow(2, height - ceng) - 1;
                mod = start * 2 + 2;

            }
        }

        return result;
    }

    //递归计算树的最大深度
    public static int getHeight(TreeNode root) {
        if (root == null) {
            return 0;
        } else {
            int lh = getHeight(root.left);
            int rh = getHeight(root.right);
            return Math.max(lh, rh) + 1;
        }
    }