题面:
在一个 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;
}
}