题目:
请实现两个函数,分别用来序列化和反序列化二叉树
分析:
序列化就是将一颗二叉树根据前序遍历序列化成字符串,反序列化就是根据一颗二叉树的先序遍历顺序将它重建成一颗二叉树。注意stoi()函数将字符串中第一个数字串转为整型,即遇到逗号啥的就停止,to_string()将一个整型转为字符串对象.
class Solution {
public:
//序列化
//将一颗二叉树序列化成字符串
//(先序)将序列化的字符串放在字符串对象tmp中,元素之间用逗号相互隔开,遇到空用#代替
void _Serialize(TreeNode* root, string& tmp)
{
if(root==nullptr)//遇到空,字符串加等#,
{
tmp+="#,";
return;
}
//说明root不是空
tmp+=to_string(root->val);//将该节点中的元素取出来转为字符串,加等到tmp上
tmp+=',';//再在后面加等一个逗号(元素之间用逗号隔开)
//再去递归左右子树
_Serialize(root->left, tmp);
_Serialize(root->right,tmp);
}
char* Serialize(TreeNode *root) {
//序列化
if(root==nullptr)
return nullptr;
string tmp;
_Serialize(root, tmp);//将序列化结果放在tmp中
char* ret=new char[tmp.size()+1];//开一个和tmp一样大的空间
strcpy(ret, tmp.c_str());//将获取到的序列拷贝到ret中
return ret;
}
//反序列化
//拿到一个二叉树先序遍历序列(重建二叉树)
//每次拿到字符串中的第一个字符,如果是'#',就跳过#和,否则就拿着这个元素构建一个节点,并且更新字符串
//然后递归构建左子树,再构建右子树(整个过程中,用来构建二叉树的序列str是一直变化的,每构建一个节点)
TreeNode* Deserialize(char *str)
{
if(str==nullptr)
return nullptr;
string tmp(str);//用字符串序列构建一个字符串对象传过去
return _Deserialize(tmp);
}
TreeNode* _Deserialize(string &str) {
if(str.empty())
return nullptr;
if(str[0]=='#')//则跳过#和,将str更新成从下标为2开始的字符串
{
str=str.substr(2);
return nullptr;//返回nullptr,即连上空
}
//来到这说明str[i]是元素,则拿着元素构建一个节点
TreeNode* root=new TreeNode(stoi(str));//stoi将取到有效数字作为字符串,遇到其它停止(即取到第一个有效数字串)
str=str.substr(str.find(',')+1);//跳过上一个结点后面的逗号,将str更新为下一个数字串开始
//递归构建左右子树
root->left=_Deserialize(str);
root->right=_Deserialize(str);
return root;//返回根结点
}
};