问题
给你二叉树的根节点 root
,请你采用前序遍历的方式,将二叉树转化为一个由括号和整数组成的字符串,返回构造出的字符串。
空节点使用一对空括号对 "()"
表示,转化后需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
示例 1:
输入:root = [1,2,3,4] 输出:"1(2(4))(3)" 解释:初步转化后得到 "1(2(4)())(3()())" ,但省略所有不必要的空括号对后,字符串应该是"1(2(4))(3)" 。
示例 2:
输入:root = [1,2,3,null,4] 输出:"1(2()(4))(3)" 解释:和第一个示例类似,但是无法省略第一个空括号对,否则会破坏输入与输出一一映射的关系。
提示:
- 树中节点的数目范围是
[1, 104]
-1000 <= Node.val <= 1000
解析
从示例1和示例2可以看出,当右为空时,可以省略括号;当左右同时为空时,可以省略括号;但是当左为空时,不能随便省略,因为不能判断是否右为空。
代码
class Solution {
public:
string tree2str(TreeNode* root) {
string str;
if (root == nullptr)
return str;
str += to_string(root->val); //to_string()函数
if (root->right || root->left)
{
str += "("; //不能用push_back
str += tree2str(root->left);
str += ")";
}
if (root->right)
{
str += "(";
str += tree2str(root->right);
str += ")";
}
return str;
}
};
采用先序,先遍历左树,当左右同时为空时,可以省略括号(此时树也是空)。对立事件就是左不为空,或者右不为空。
遍历右树,右为空时,可以省略括号。对立事件就是右不为空,不能省略括号。
注意点
1.使用to_string函数转换成字符串
2.+=操作符(string最好用的push方法)可以+= sting对象
而push_back只允许char字符
3. str += tree2str(root->left);应该放在if之内,因为左右树为空时,不需要push到字符串中