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

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

思路:

如果父节点为空,则直接返回空串

如果父节点不为空,那么就把父节点的值加到字符串s中

再判断左右子树是否为空:

左空、右不空:返回s+"()"+"("+tree2str(t->right)+")"

左空、右空:返回s

左不空、右空:返回s+"("+tree2str(t->right)+")"

左不空、右不空:返回s++"("+tree2str(t->left)+")"+"("+tree2str(t->right)+")"

写代码时逻辑可以优化一下,变成先后判断,优化后代码如下:

/**
 * 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)
        {
            string s;
            s+=to_string(t->val);
            if(t->left)
            {
                s+="(";
                s+=tree2str(t->left);
                s+=")";
            }
            else
            {
                if(t->right)
                    s+="()";
            }
            if(t->right)
            {
                s+="(";
                s+=tree2str(t->right);
                s+=")";
            }
            return s;
        }
        else
            return "";
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_40673608/article/details/83119378