题目描述:
你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。
空节点则用一对空括号 “()” 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
示例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)”
解释: 和第一个示例相似,
除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
string solve(TreeNode* root){
if(root==nullptr)return "";//根节点为空,不输出
stringstream ss;
ss<<root->val;
string num;
ss>>num;
if(root->left==nullptr&&root->right==nullptr)return num; //该根节点无左右子树,直接返回改节点值
string left = solve(root->left);
if(root->right!=nullptr){ //右子树为空时,直接省略,非空时才显示
string right = solve(root->right);
return num + "(" + left + ")" + "(" + right + ")";//左右子树用左右括号括起来
}
return num+"("+left+")";//右子树为空时只显示左子树,(若左子树为空,则表示为"()")
}
string tree2str(TreeNode* root) {
if(root==nullptr)return "()";
return solve(root);
}
};
递归法。
对于每个子树(左子树和右子树),都用左右括号括起来。根节点本身不用阔号。题目要求简化,当右子树为空时,直接省略不写(即把“()”省略),在递归诗要先判断右子树是否为空再继续递归。