剑指offer - 61 -- 序列化二叉树 - C++

题目描述

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

 

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

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

例如，我们可以把一个只有根节点为1的二叉树序列化为"1,"，然后通过自己的函数来解析回这个二叉树

示例1

输入

{8,6,10,5,7,9,11}

返回值

{8,6,10,5,7,9,11}

题解

层序遍历实现

class Solution {
public:
    char* Serialize(TreeNode *root) {    
        queue<TreeNode*> q;
        if(!root)
            return "#";
        q.push(root);
        string s="";
        while(!q.empty())
        {
            TreeNode *node = q.front();
            q.pop();
            if(node == nullptr){
                s += "#";
            }
            s += to_string(node->val);
            s += "!";
            q.push(node->left);
            q.push(node->right);
        }
        char *ret = new char[s.length()+1];
        strcpy(ret, s.c_str());
        return ret;
    }
    TreeNode* Deserialize(char *str) {
        if(str == nullptr)
            return nullptr;
        string s(str);
        if(str[0] == '#')
            return nullptr;
        //构造树的头结点
        queue<TreeNode* > nodes;
        TreeNode* ret = new TreeNode(atoi(s.c_str()));
        s = s.substr(s.find_first_of('!') + 1);
        nodes.push(ret);
        while(!nodes.empty() && !s.empty())
        {
            TreeNode *node = nodes.front();
            nodes.pop();
            if(s[0] == '#')
            {
                node->left = nullptr;
                s = s.substr(2);
            }
            else
            {
                node->left = new TreeNode(atoi(s.c_str()));
                nodes.push(node->left);
                s= s.substr(s.find_first_of('!') + 1);
            }
            if(s[0] == '#')
            {
                node->right = nullptr;
                s = s.substr(2);
            }
            else
            {
                node->right = new TreeNode(atoi(s.c_str()));
                nodes.push(node->right);
                s = s.substr(s.find_first_of('!')+1);
            }
        }
        return ret;
    }
};