例题-序列化二叉树

题目描述

请实现两个函数，分别用来序列化和反序列化二叉树

二叉树的序列化是指：把一棵二叉树按照某种遍历方式的结果以某种格式保存为字符串，从而使得内存中建立起来的二叉树可以持久保存。序列化可以基于先序、中序、后序、层序的二叉树遍历方式来进行修改，序列化的结果是一个字符串，序列化时通过 某种符号表示空节点（#），以 ！ 表示一个结点值的结束（value!）。

二叉树的反序列化是指：根据某种遍历顺序得到的序列化字符串结果str，重构二叉树。

原理

序列化二叉树

递归遍历二叉树的节点，空节点使用#代替，节点之间使用！隔开，返回字符串

反序列化二叉树

设置序号index，将字符串根据逗号分割为数组，根据index的值来设置树节点的val，如果节点的值为#，则返回空的树节点。

代码实现

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) 
    {       //序列化
        if(!root)
            return (char*)"#";
        string res = to_string(root->val);
        res += "*";
        res += Serialize(root->left);
        res += Serialize(root->right);
        char* resBuf = new char[res.size()];
        strcpy(resBuf,res.c_str());
        return resBuf;
    }
    TreeNode* deser(string& str,int& pos)
    {
        if(str[pos] == '#')
        {
            ++pos;
            return nullptr;
        }
        auto sum = 0;
        for(;pos < str.size();++pos)
        {
            if(str[pos]!='*')
            {
                sum *= 10;
                sum += str[pos]-'0';
            }
            else
            {
                ++pos;
                break;
            }
        }
        TreeNode* res = new TreeNode(sum);
        res->left = deser(str,pos);
        res->right = deser(str,pos);
        return res;
    }
    TreeNode* Deserialize(char *str) 
    {
        string nStr = str;
        int pos = 0;
        return deser(nStr, pos);
    }
};