题目描述
请实现两个函数,分别用来序列化和反序列化二叉树
思路
首先来解释一下什么是序列化和反序列化二叉树。如下图所示的二叉树。
序列化二叉树就是以先序、中序、后序、层序任意一种遍历二叉树的方法将二叉树遍历,结点值之间以逗号分隔,空指针以井号代替。以先序序列为例,序列化上图二叉树的结果就是[1,2,4,#,#,5,#,#,3,#,#]。
反序列化二叉树就是给定如上所示的序列,以序列化时的遍历方法(先序、中序、后序、层序)将序列还原回二叉树。
编程的思路就按定义来实现就好,我这里用的是先序遍历。在序列化二叉树时注意要在最后添加‘\0’为序列结束的标识。
代码(c++)
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
void preorder(TreeNode *root,string &str){
if(root==NULL){
str+='#';
str+=',';
return;
}
str+=to_string(root->val);
str+=',';
preorder(root->left,str);
preorder(root->right,str);
}
char* Serialize(TreeNode *root) {
if(root==NULL) return NULL;
string str;
preorder(root,str);
char *res=new char[str.length()];
for(int i=0;i<str.length();i++){
res[i]=str[i];
}
res[str.length()-1]='\0';
return res;
}
void buildTree(char* &str, TreeNode* &root){
if(*str=='\0') return;
if(*str==',') str+=1;
if(*str=='#'){
root=NULL;
str+=1;
return;
}
int temp=0;
while(*str!=','&&*str!='\0'){
temp=temp*10+(*str-'0');
str+=1;
}
root=new TreeNode(temp);
buildTree(str,root->left);
buildTree(str,root->right);
}
TreeNode* Deserialize(char *str) {
if(str==NULL) return NULL;
TreeNode* root;
buildTree(str, root);
return root;
}
};