你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。
空节点则用一对空括号 "()" 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
示例 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)"
解释: 和第一个示例相似,
除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。
class Solution {
public:
void change(TreeNode* root, string& s){
if (root->left != nullptr){
s += '(' + to_string(root->left->val);
change(root->left, s);
s += ')';
}
if (root->right != nullptr){
if (root->left == nullptr)
s += "()";
s += '(' + to_string(root->right->val);
change(root->right, s);
s += ')';
}
}
string tree2str(TreeNode* root) {
string res;
if (root == nullptr)
return "";
res += to_string(root->val);
TreeNode* p = root;
change(p, res);
return res;
}
};