606. 根据二叉树创建字符串

你需要采用前序遍历的方式,将一个二叉树转换成一个由括号和整数组成的字符串。

空节点则用一对空括号 "()" 表示。而且你需要省略所有不影响字符串与原始二叉树之间的一对一映射关系的空括号对。

示例 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)"

解释: 和第一个示例相似,
除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-string-from-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

/**
 * 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:
    void dfs(string &s,TreeNode*t){
        if(!t)return;
        s+=to_string(t->val);
        s.insert(s.end(),1,'(');
        dfs(s,t->left);
        s.insert(s.end(),1,')');
        s.insert(s.end(),1,'(');
        dfs(s,t->right);
        s.insert(s.end(),1,')');
    }
    string tree2str(TreeNode* t) {
        string s;
        dfs(s,t);
        for (int i = 0; i + 4 <= s.size (); i ++) // "()()"-->""
            if(s.substr(i,4)=="()()")
               s.erase(i,4);
        for (int i = 0; i + 3 <= s.size (); i ++) // ")()"-->")"
            if(s.substr(i,3)==")()")
               s.erase(i+1,2);    
        return s;
    }
};

 

posted @ 2020-11-25 15:45  XXXSANS  阅读(116)  评论(0编辑  收藏  举报