leetcode606. 根据二叉树创建字符串

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

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

示例:
 输入:二叉树:[1, 2, 3, null, 4]
在这里插入图片描述
 输出:"1(2()(4))(3)"

思路:
题目要求我们省略所有不影响字符串与原始二叉树之间一对一映射关系的空括号对,也就是只保留必要的空括号对。

题目要求采用前序遍历的方式,因此对于一棵不为空的二叉树,我们首先应该将该二叉树的根结点值放入到字符串当中,而对于该树的左孩子,只有该树的属于下面两种情况之一时,才有必要将该树的左孩子的值放入到字符串当中。
在这里插入图片描述
因为对于情况二来说,如果不将其左孩子的值(空括号对)放入字符串中,我们将不能区分结点3是结点1的左孩子还是右孩子。而对于二叉树的右孩子,只要右孩子不为空我们就需要将右孩子的值放入字符串。

因此,字符串创建的过程如下:

  1. 若二叉树不为空,则先根结点的值放入字符串。
  2. 若左子树不为空,或者左子树为空但右子树不为空,则需要将左子树的值放入字符串。
  3. 若右子树不为空,则需要将右子树的值放入字符串。

其中,操作根结点左子树和右子树的规则也是如此。

代码如下:

/**
 * 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) {
		if (root == nullptr) //空树
		{
			return string(); //返回空字符串
		}
		string str;
		//1、先根结点的值放入字符串
		str += to_string(root->val);
		
		//2、若左子树不为空,或者左子树为空但右子树不为空,则需要将左子树的值放入字符串
		if (root->left || root->right)
		{
			str += "(";
			str += tree2str(root->left); //操作左子树的规则相同
			str += ")";
		}

		//3、若右子树不为空,则需要将右子树的值放入字符串
		if (root->right)
		{
			str += "(";
			str += tree2str(root->right); //操作右子树的规则相同
			str += ")";
		}
		return str; //返回字符串
	}
};

该代码虽然能够成功创建字符串,但该代码有一个致命的问题:当所给二叉树的结构比较大时,因为该函数使用的是传值返回,所以在创建字符串时会存在大量的string深拷贝。
为了解决该问题,我们可以选择封装一个子函数,子函数的传参使用引用传参,这时便能避免进行string的深拷贝。

优化代码如下:

/**
 * 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){
		string str;
		_tree2str(root, str); //调用子函数,子函数采用引用传参
		return str; //返回字符串
	}
	void _tree2str(TreeNode* root, string& str) {
		if (root == nullptr) //空树
		{
			return; //直接返回
		}
		//1、先根结点的值放入字符串
		str += to_string(root->val);

		//2、若左子树不为空,或者左子树为空但右子树不为空,则需要将左子树的值放入字符串
		if (root->left || root->right)
		{
			str += "(";
			_tree2str(root->left, str); //操作左子树的规则相同
			str += ")";
		}

		//3、若右子树不为空,则需要将右子树的值放入字符串
		if (root->right)
		{
			str += "(";
			_tree2str(root->right, str); //操作右子树的规则相同
			str += ")";
		}
	}
};
  • 12
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 9
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

2021dragon

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值