leetcode 606. 根据二叉树创建字符串(Construct String from Binary Tree)

题目描述:

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

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

示例 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(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    string toString(int num){
        bool neg = false;
        if(num == 0){
            return "0";
        }else if(num < 0){
            num = -num;
            neg = true;
        }
        string res = "";
        while(num){
            res = char('0' + num%10) + res;
            num /= 10;
        }
        if(neg){
            return '-' + res;
        }else{
            return res;
        }
    }
    
    
    string tree2str(TreeNode* t) {
        if(!t){
            return "";
        }else if(!t->left && !t->right){
            return toString(t->val);
        }else{
            string res = "";
            res = toString(t->val);
            res += '(' + tree2str(t->left) + ')';
            if(t->right){
                res += '(' + tree2str(t->right) + ')';
            }
            return res;
        }
    }
};

猜你喜欢

转载自www.cnblogs.com/zhanzq/p/10600728.html