leedcode 655 輸出二叉樹

在一個 m*n 的二維字符串數組中輸出二叉樹,並遵守以下規則:

行數 m 應當等於給定二叉樹的高度。
列數 n 應當總是奇數。
根節點的值(以字符串格式給出)應當放在可放置的第一行正中間。根節點所在的行與列會將剩餘空間劃分爲兩部分(左下部分和右下部分)。你應該將左子樹輸出在左下部分,右子樹輸出在右下部分。左下和右下部分應當有相同的大小。即使一個子樹爲空而另一個非空,你不需要爲空的子樹輸出任何東西,但仍需要爲另一個子樹留出足夠的空間。然而,如果兩個子樹都爲空則不需要爲它們留出任何空間。
每個未使用的空間應包含一個空的字符串""。
使用相同的規則輸出子樹。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/print-binary-tree
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

9,20 16:40 ~ 17:12
這題目基本思路:

  1. 先算數的深度,通過深度得到m,n
  2. 採用廣度遍歷給m * n矩陣賦值
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def get_max_depth(self,node,d):
        if node.left != None:
            self.get_max_depth(node.left,d+1)
        if d > self.depth:
            self.depth = d
        if node.right != None:
            self.get_max_depth(node.right,d+1)
    
    def print_key(self,node,d,i,j):
        mid = (i + j) /2
        self.ret[d-1][mid] = str(node.val)
        if node.left != None:
            self.print_key(node.left,d+1,i,mid-1)
        if node.right != None:
            self.print_key(node.right,d+1,mid+1,j)
        
    def printTree(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[str]]
        """
        self.depth = 0
        self.get_max_depth(root,1)
        m = self.depth
        n = sum([2**i for i in range(m)])
        self.ret = []
        for i in range(m):
            tmp = []
            for j in range(n):
                tmp.append("")
            self.ret.append(tmp)
        self.print_key(root,1,0,n-1)
        return self.ret
        
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章