leetcode 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)"

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

代码:

方法一:自己码的

class Solution {
public:
    string tree2str(TreeNode* t) {
        string str;
        if(t==NULL)return str;
        pre(t,str);
        str.erase(str.begin());//此方法最后字符串首尾分别为“(”,“)”,应该删除
        str.erase(str.end()-1);
        return str;
    }
    void pre(TreeNode * t,string &str){
        if(t==NULL) return ;
        str+="("+to_string(t->val);
        if(!t->left&&t->right)//加()的情况
            str+="()";
        else
            pre(t->left,str);
        pre(t->right,str);
        str+=")";
    }
};

方法二:大佬代码。

class Solution {
public:
    string tree2str(TreeNode* t) {
        if(t==nullptr) return "";
        string res;
        res+=to_string(t->val);
        if(t->left) res+="("+tree2str(t->left)+")";
        else if( t->right) res+="()";
        if(t->right) res+="("+tree2str(t->right)+")";
        return res;
    }
};
如此简洁,我还是差远了

猜你喜欢

转载自blog.csdn.net/weixin_42130471/article/details/80342837