题目
原题链接
你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。
空节点则用一对空括号 “()” 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。
示例 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)"
解释: 和第一个示例相似,
除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。
思路
这个shi一样的题目描述……算了我还是先讲讲题目的意思吧。
研究一下题目,我们发现,其实这道题的答案很明显是一个递归的关系,总体来说可以将答案的字符串理解为 root->val(左子树)(右子树)
。这个应该很好理解,但是题目中还有省略掉一些括号,这是什么意思呢?以示例一为例:
输入: 二叉树: [1,2,3,4]
1
/ \
2 3
/
4
输出: "1(2(4))(3)"
解释: 原本将是“1(2(4)())(3())”,
在你省略所有不必要的空括号对之后,
它将是“1(2(4))(3)”。
为啥呢?注意该树的2这个节点,它的右子树不存在,所以,要省略掉4后的一对括号。
然后再看3,他没有左右子树,所以后面的两个括号都要省略。
那假设3只有右子树,没有左子树呢?该如何表示?——3()(val)
;
有人就问了,为什么3后面的那对空括号不省略呢?——如果把他删掉,就体现不出(val)
是右子树了。
其实,搞懂了题意,接下来的就很简单了,以下是代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
string ans;
string tree2str(TreeNode* t) {
if(!t) return "";
ans += to_string(t -> val);
if(!t -> left && !t -> right) return ans;
ans += '(';
if(t -> left) digit(t -> left);
ans += ')';
if(t -> right) {
ans += '(';
digit(t -> right);
ans += ')';
}
return ans;
}
void digit(TreeNode* root) {
if(!root) { ans += "()"; return ; }
if(!root -> left && !root -> right)
{ans += to_string(root -> val); return ; }
ans += to_string(root -> val);
ans += '(';
if(root -> left) {
digit(root -> left);
}
ans += ')';
if(root -> right) {
ans += '(';
digit(root -> right);
ans += ')';
}
}
};