题目描述
请实现两个函数,分别用来序列化和反序列化二叉树
这个题的难点是强行给了个C语言的返回值让写,因此在转化的过程中非常的麻烦。
几个注意点:str.c_str()得到的是const char* 类型的,若想返回char*的可以使用const_cast去掉const属性。
注意注释的地方,序列化的过程和反序列化正好是相反的,可能反序列化不太好像,只要理解是递归过程就好一些了。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
char* Serialize(TreeNode *root) {
string str=Serializestr(root);
return const_cast<char*>(str.c_str());
}
TreeNode* Deserialize(char *str) {
string input(str);
int begin=0;
return Deserializestr(input,begin);
}
TreeNode* Deserializestr(string str,int &begin) { //这个地方必须是要引用的,因为每次递归位置都需要改变,这个特别需要注意。
if(str[begin]=='#')
{
++begin;
return NULL;
}
int beg=begin;
while(str[begin]!='!')
++begin;
++begin;
int val=atoi(str.substr(beg,begin-beg).c_str());
TreeNode *head=new TreeNode(val);
head->left=Deserializestr(str,begin);
head->right=Deserializestr(str,begin);
return head;
}
string Serializestr(TreeNode *root) {
if(root==NULL)
return "#";
string res{};
res+=to_string(root->val)+"!";
res+=Serializestr(root->left);
res+=Serializestr(root->right);
return res;
}
};