【树】606. 根据二叉树创建字符串

题目:

 

解答:

方法一:递归

我们可以使用递归的方法得到二叉树的前序遍历。在递归时,根据题目描述,我们需要加上额外的括号,会有以下4中情况。

  • 如果当前节点有两个孩子,那我们递归时,需要在两个孩子的结果外加上一层括号;
  • 如果当前节点没有孩子,那我们不需要在节点后面加上任何括号;

  • 如果当前节点只有左孩子,那我们在递归时,只需要在左孩子的结果外加上一层括号,而不需要给右孩子加上任何括号;

  • 如果当前节点只有右孩子,那我们在递归时,需要先加上一层空括号()表示左孩子为空,再对右孩子进行递归,并在结果外加上一层括号;

 考虑完上面4种情况,我们就可以得到最终的字符串。

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12      string tree2str(TreeNode *root) 
13      {
14         if(root == NULL)
15             return "";
16 
17         std::string str = std::to_string(root->val);
18 
19         if(root->left == NULL && root->right == NULL)
20             return str + "";
21  
22         if(root->right == NULL)
23             return str + "("+tree2str(root->left)+")";
24 
25         return str + "("+tree2str(root->left)+")("+tree2str(root->right)+")";   
26     }
27 };

猜你喜欢

转载自www.cnblogs.com/ocpc/p/12821837.html