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

链接

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)"
解释: 和第一个示例相似，
除了我们不能省略第一个对括号来中断输入和输出之间的一对一映射关系。

提示

题目说的很模糊，其实意思就是：

将子节点的值用()来包裹。对于二叉树[root,left,right]，转换成root(left)(right)。如果只有left为空节点，则输出root()(right)；如果只有right为空节点则可以忽略右节点的()，输出为root(left)。

思路一（递归）

见代码注释

C++ Code

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    string tree2str(TreeNode* root) {
        //1. 根节点为空
        if(root==nullptr) return "";
        //2. 左右节点均为空
        if(root->left==nullptr && root->right==nullptr) return to_string(root->val);
        //3. 左节点为空 右节点不为空
        else if(root->left==nullptr && root->right!=nullptr) return to_string(root->val)+"()"+"("+tree2str(root->right)+")";
        //4. 左节点不为空 右节点为空
        else if(root->left!=nullptr && root->right==nullptr) return to_string(root->val)+"("+tree2str(root->left)+")";
        //5. 左右节点均非空
        else return to_string(root->val)+"("+tree2str(root->left)+")"+"("+tree2str(root->right)+")";

    }
};

其中，左节点为空右节点不为空的情况和左右节点均非空的情况可以合并。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    string tree2str(TreeNode* root) {
        //1. 根节点为空
        if(root==nullptr) return "";
        //2. 左右节点均为空
        if(root->left==nullptr && root->right==nullptr) return to_string(root->val);
        //3. 左节点为空 右节点不为空
        //注意第3种情况被注释掉了，和第5种一样
        //else if(root->left==nullptr && root->right!=nullptr) return to_string(root->val)+"()"+"("+tree2str(root->right)+")";
        //4. 左节点不为空 右节点为空
        else if(root->left!=nullptr && root->right==nullptr) return to_string(root->val)+"("+tree2str(root->left)+")";
        //5. 左右节点均非空
        else return to_string(root->val)+"("+tree2str(root->left)+")"+"("+tree2str(root->right)+")";

    }
};

思路二（迭代）

待更。