【leetcode刷题】T145-根据二叉树创建字符串

示例 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.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def tree2str(self, t):
        """
        :type t: TreeNode
        :rtype: str
        """
        self.res = ''
        if not t:
            return ''
        self.traval(t)
        return self.res[1:-1]

    def traval(self, t):
        self.res += '(' + str(t.val)
        if t.left:
            self.traval(t.left)
        elif t.right:
            self.res += '()'
        if t.right:
            self.traval(t.right)
        self.res += ')'

/**
 * 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 tree2str(TreeNode* t) {
        if(!t)
            return "";
        string str="";
        traval(t, str);
        return str.substr(1, str.size()-2);
    }

    void traval(TreeNode* t, string& str){
        str.append("(");
        str.append(to_string(t->val));
        if(t->left)
            traval(t->left, str);
        else
            if(t->right)
                str.append("()");
        if(t->right)
            traval(t->right, str);
        str.append(")");
    }
};