题目链接:https://leetcode-cn.com/problems/serialize-and-deserialize-bst/
题目描述
序列化是将数据结构或对象转换为一系列位的过程,以便它可以存储在文件或内存缓冲区中,或通过网络连接链路传输,以便稍后在同一个或另一个计算机环境中重建。
设计一个算法来序列化和反序列化二叉搜索树。 对序列化/反序列化算法的工作方式没有限制。 您只需确保二叉搜索树可以序列化为字符串,并且可以将该字符串反序列化为最初的二叉搜索树。
编码的字符串应尽可能紧凑。
注意:不要使用类成员/全局/静态变量来存储状态。 你的序列化和反序列化算法应该是无状态的。
思路
对于二叉搜索树而言,左子节点<根节点<右子节点;根据这个性质,我们用先序遍历序列就可以进行重建了,并且不需要存储空指针。
如果是一个普通二叉树,需要前序和中序,或者后序和中序两个序列,才能唯一确定二叉树结构,而二叉搜索树,由于各元素之间有左子节点小于父节点再小于右子节点的性质,因此用一个前序排列就可以确定出唯一结构。
代码
class Codec {
public:
// Encodes a tree to a single string.
string serialize(TreeNode* root) {
if(!root) return "";
string ret = to_string(root->val);
ret.push_back(',');
ret += serialize(root->left);
ret += serialize(root->right);
return ret;
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
vector<int> nums;
// 切割数据
while(!data.empty()){
int pos = data.find(',');
int val = stoi(data.substr(0,pos));
nums.push_back(val);
data = data.substr(pos+1);
}
return buildTree(nums,0, nums.size() -1);
}
// 从先序遍历序列重建二叉搜索树
TreeNode* buildTree(vector<int> &nums, int begin, int end) {
if(begin > end || begin < 0 || end >= nums.size())
return nullptr;
TreeNode *root = new TreeNode(nums[begin]);
int lh = begin + 1;
while(lh <= end && nums[lh] < root->val)
lh ++; // 第一个大于root值的坐标
root->left = buildTree(nums, begin + 1, lh-1); // 递归构建左子树
root->right = buildTree(nums, lh, end); // 递归构建右子树
return root;
}
};