- 输出二叉树
在一个 m*n 的二维字符串数组中输出二叉树,并遵守以下规则:
行数 m 应当等于给定二叉树的高度。
列数 n 应当总是奇数。
根节点的值(以字符串格式给出)应当放在可放置的第一行正中间。根节点所在的行与列会将剩余空间划分为两部分(左下部分和右下部分)。你应该将左子树输出在左下部分,右子树输出在右下部分。左下和右下部分应当有相同的大小。即使一个子树为空而另一个非空,你不需要为空的子树输出任何东西,但仍需要为另一个子树留出足够的空间。然而,如果两个子树都为空则不需要为它们留出任何空间。
每个未使用的空间应包含一个空的字符串""。
使用相同的规则输出子树。
示例 1:
输入:
1
/
2
输出:
[["", “1”, “”],
[“2”, “”, “”]]
示例 2:
输入:
1
/
2 3
4
输出:
[["", “”, “”, “1”, “”, “”, “”],
["", “2”, “”, “”, “”, “3”, “”],
["", “”, “4”, “”, “”, “”, “”]]
示例 3:
输入:
1
/
2 5
/
3
/
4
输出:
[["", “”, “”, “”, “”, “”, “”, “1”, “”, “”, “”, “”, “”, “”, “”]
["", “”, “”, “2”, “”, “”, “”, “”, “”, “”, “”, “5”, “”, “”, “”]
["", “3”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”]
[“4”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”, “”]]
注意: 二叉树的高度在范围 [1, 10] 中。
思路很简单:
看代码就能理解
先把数组长度求出来,然后求树的深度。都是递归思想
然后递归的向数组中填入数据!
不知道他的树的高度还有深度是从下标几开始,所以可能存在数组越界的情况
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def printTree(self, root: TreeNode) -> List[List[str]]:
def get_hight(root: TreeNode):
if not root:
return 0
return max(get_hight(root.left), get_hight(root.right)) + 1
# 递归填写,类似于二分查找
def fill(root: TreeNode, hight: int, left: int, right: int):
if not root:
return
mid = (right + left) // 2 -1
ret[hight][mid] = str(root.val)
fill(root.left, hight+1, left, mid-1)
fill(root.right, hight+1, mid+1, right)
higth = get_hight(root)
width = pow(higth, 2) - 1
ret = [["" for _ in range(width)]for _ in range(higth)]
fill(root, 0, 0, pow(higth, 2) - 1)
return ret