题目描述
你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。
空节点则用一对空括号 “()” 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
输入: 二叉树: [1,2,3,4]
输出: “1(2(4))(3)”
解释: 原本将是“1(2(4)())(3())”,
在你省略所有不必要的空括号对之后,
它将是“1(2(4))(3)”。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-string-from-binary-tree/
解题思路
核心就是先序遍历,打印节点的时候加上()就行了,那么关键就是什么时候可以省略,什么时候不能省略:
左右子树都为空的时候,此时可以省略左右子树的();
左子树不为空,右子树为空,也可以省略右子树的空();
左子树为空,右子树不为空,不能省略空();
这里我们使用了一个辅助函数,我们在添加每一个节点的时候,就先把(,加入,然后再加入该节点的左孩子,右孩子加入,再把)写入。
代码
class Solution {
StringBuilder result = null;
public String tree2str(TreeNode t) {
result = new StringBuilder();
if(t == null){
return "";
}
Swep(t);
result.deleteCharAt(0);
result.deleteCharAt(result.length()-1);
return result.toString();
}
public void Swep(TreeNode root){
if(root == null){
return;
}
result.append("(");
result.append(root.val);
Swep(root.left);
if(root.left == null && root.right != null){
result.append("()");
}
Swep(root.right);
result.append(")");
}
}