题目描述
你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。
空节点则用一对空括号 “()” 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
输入: 二叉树: [1,2,3,4]
1
/ \
2 3
/
4
输出: "1(2(4))(3)"
解释: 原本将是“1(2(4)())(3())”,在你省略所有不必要的空括号对之后,它将是“1(2(4))(3)”。
输入: 二叉树: [1,2,3,null,4]
1
/ \
2 3
\
4
输出: "1(2()(4))(3)"
解释: 和第一个示例相似,除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。
算法思路
二叉树天然适合递归。
class Solution:
def tree2str(self, t: TreeNode) -> str:
if not t:return ''
if not (t.left or t.right):return str(t.val)
l=self.tree2str(t.left) #if t.right else ''
r='('+str(self.tree2str(t.right))+')' if t.right else ''
return '%d(%s)%s'%(t.val,l,r)
!这里只着重一点,就是不存在右节点,可以直接省略'()'
,但是左节点为空,不能省略括号,需要用来保存映射关系。