655. 输出二叉树

题目描述：

在一个 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] 中。

个人思路：

根据题目描述不难看出，这题也可以使用递归来完成。

观察示例3：

首先考虑root节点，也就是数值为1的节点。其位于完整matrix的第一行的中央位置。而root所在行以及所在列将完整matrix分为左下角以及右下角部分，也就是图中所示红线划去的元素以外的部分。此时，我们得到两个新的matrix——left以及right，不难发现，新的matrix和原本完整的matrix没有本质差别，只是“小了”一些，很明显，递归可以很好地适应这种要求。

确定好使用递归方法后，需要首先构造出完整matrix，即需要知道该matrix的高和宽。具体规律是，高为树的depth，而宽为2 ^ depth - 1，这一点可以观察示例3，从最后一行往上看，比较易于发现。

二叉树递归算法遵循一个原则：递归主体只处理root节点，剩下来的部分交给左右孩子节点的递归程序即可。

另外代码中由于使用了二维数组切片，因此使用了numpy数组。但有个坑需要注意，numpy的字符串数组默认限定字符长度为1，因此需要将numpy数组类型设置为object。

具体代码如下：

# Definition for a binary tree node.
import numpy as np
from typing import List

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]]:
        depth = self.get_depth(root)
        width = 2 ** depth - 1
        # 使用numpy数组，方便切片，但需要注意numpy字符数组默认限定长度为1，因此改用object，从而可以赋值为任意长度字符串！大坑！
        matrix = np.array([[''] * width for _ in range(depth)], dtype=object)  
        
        def helper(root, matrix):
            if not root:
                return matrix
            mid = len(matrix[0]) // 2
            matrix[0][mid] = str(root.val)  # root节点赋值操作
            left_part = matrix[1:, :mid]
            right_part = matrix[1:, mid + 1:]
            matrix[1:, :mid] = helper(root.left, left_part)  # 递归更新left
            matrix[1:, mid + 1:] = helper(root.right, right_part)  # 递归更新right
            return matrix
        
        return list(helper(root, matrix))
    
    def get_depth(self, root):  # 递归获取深度
        if not root:
            return 0
        
        def dfs(root):
            if not root:
                return 0
            return max(dfs(root.left), dfs(root.right)) + 1
        
        return dfs(root)