Leetcode 449. 序列化和反序列化二叉搜索树 C++

Leetcode 449. 序列化和反序列化二叉搜索树

题目

序列化是将数据结构或对象转换为一系列位的过程，以便它可以存储在文件或内存缓冲区中，或通过网络连接链路传输，以便稍后在同一个或另一个计算机环境中重建。

设计一个算法来序列化和反序列化二叉搜索树。 对序列化/反序列化算法的工作方式没有限制。 您只需确保二叉搜索树可以序列化为字符串，并且可以将该字符串反序列化为最初的二叉搜索树。

编码的字符串应尽可能紧凑。

注意：不要使用类成员/全局/静态变量来存储状态。 你的序列化和反序列化算法应该是无状态的。

题解

利用栈的先序遍历方式反序列化二叉搜索树
我们先序遍历获取序列化，其中若为空树，则用n记录，此外结点与结点间用"."分割。
对于反序列化，我们先尽可能的创建左子树，当序列为n，则说明无左子树，我们就创建右子树。
详细过程见代码

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Codec {
public:

    // Encodes a tree to a single string.
    string serialize(TreeNode* root) {		//先序遍历进行序列化
        if(root ==NULL) return "n.";
        string ans;
        ans += to_string(root->val)+".";
        ans += serialize(root->left);
        ans += serialize(root->right);
        return ans;
    }
    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        int n=data.length();
        int i=0;
        if(data[i] == 'n')  return nullptr;
        stack<TreeNode*> sp;
        bool isF = false;
        int now=0;
        if(data[i] == '-'){		//获取根节点的val
            isF = true;
            i++;
        }  
        while(data[i]!='.'){
            now = now*10 + (data[i]-'0');
            i++;
        }
        if(isF) now = -now;
        TreeNode* root = new TreeNode(now);		//创建根节点
        sp.push(root);
        while(!sp.empty()){
            if(data[i] == '.')  i++;
            TreeNode* t = sp.top();
            if(data[i] =='n' || t->left!=NULL){		//前者说明当前结点可能无左子树，后者说明需要创建右子树
                if(t->left==NULL){			//无左子树的情况
                    t->left = nullptr;
                    i+=2;
                }
                if(data[i] == 'n'){			//说明无右子树
                    t->right = nullptr;
                    sp.pop();
                }else{			//创建右子树的根节点
                    isF = false;
                    now = 0;
                    if(data[i] == '-'){
                        isF = true;
                        i++;
                    }  
                    while(data[i]!='.'){
                        now = now*10 + (data[i]-'0');
                        i++;
                    }
                    if(isF) now = -now;
                    TreeNode* newT = new TreeNode(now);
                    t->right = newT;
                    sp.pop();
                    sp.push(newT);
                } 
            }else{		//创建左子树的根节点
                isF = false;
                now = 0;
                if(data[i] == '-'){
                    isF = true;
                    i++;
                }  
                while(data[i]!='.'){
                    now = now*10 + (data[i]-'0');
                    i++;
                }
                if(isF) now = -now;
                TreeNode* newT = new TreeNode(now);
                t->left = newT;
                sp.push(newT);
            }
            i += 1;
        }
        return root;
    }
};

// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/serialize-and-deserialize-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。