题目原文
你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。
空节点则用一对空括号 “()” 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
示例 1:
输入: 二叉树: [1,2,3,4]
1
/
2 3
/
4
输出: “1(2(4))(3)”
解释: 原本将是“1(2(4)())(3())”,
在你省略所有不必要的空括号对之后,
它将是“1(2(4))(3)”。
示例 2:
输入: 二叉树: [1,2,3,null,4]
1
/
2 3
\
4
输出: “1(2()(4))(3)”
解释: 和第一个示例相似,
除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。
解题思路
一、递归
- 将t.val加入到s中,若左右子树都为空,返回s;
- 若左右子树不为空,使用递归继续添加括号和值到s;
- 左子树为空添加括号,不继续递归。
class Solution:
def tree2str(self, t: TreeNode) -> str:
if not t:
return ""
return self.dfs(t)
def dfs(self, t):
s = str(t.val)
if not t.left and not t.right:
return s
if t.left:
s += "("
s += self.dfs(t.left)
s += ")"
elif not t.left:
s += "()"
if t.right:
s += "("
s += self.dfs(t.right)
s += ")"
return s
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-string-from-binary-tree