297. 二叉树的序列化与反序列化

题目

序列化是将一个数据结构或者对象转换为连续的比特位的操作，进而可以将转换后的数据存储在一个文件或者内存中，同时也可以通过网络传输到另一个计算机环境，采取相反方式重构得到原数据。

请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑，你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。

提示: 输入输出格式与 LeetCode 目前使用的方式一致，详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式，你也可以采用其他的方法解决这个问题。

示例 1：

输入：root = [1,2,3,null,null,4,5]
输出：[1,2,3,null,null,4,5]
示例 2：

输入：root = []
输出：[]
示例 3：

输入：root = [1]
输出：[1]
示例 4：

输入：root = [1,2]
输出：[1,2]

提示：

树中结点数在范围 [0, 104] 内
-1000 <= Node.val <= 1000

链接

297. 二叉树的序列化与反序列化

代码

/**
 * 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) return "#";//如果二叉树为空，则标记为#
        string str;//保存二叉树的字符串
        queue<TreeNode*> q;//队列
        q.push(root);//把根节点放入队列
        str += std::to_string(root->val) + '#';//序列化根节点的值，以#结尾
        while(q.size()){
            int cnt = q.size();
            while(cnt--){
                auto t = q.front(); q.pop();//队头出队
                if(t->left){//如果左子节点存在，则放入队列中，并且序列化节点的值
                    q.push(t->left);
                    str += std::to_string(t->left->val) + '#';
                }else str += '#';//如果左子节点不存在，则序列化该节点为#
                if(t->right){//如果右子节点存在，则放入队列中，并且序列化节点的值
                    q.push(t->right);
                    str += std::to_string(t->right->val) + '#';
                }else str += '#';//如果右子节点不存在，则序列化该节点为#
            }
        }
        return str;
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        if(data[0] == '#') return NULL;//如果序列化后的字符串为#，则二叉树为空
        int i = 0;//字符串的当前索引，从第一个位置开始遍历
        queue<TreeNode*> q;//队列
        TreeNode* root = getNode(data, i);//根节点
        q.push(root);//把根节点放入队列
        while(q.size()){
            int cnt = q.size();//当前层的节点个数
            while(cnt --){
                auto t = q.front(); q.pop();//队头出队
                auto l = getNode(data,i), r = getNode(data,i);//继续获取两个数字，返回封装好的节点
                t->left = l, t->right = r;//指定队头的左右子节点
                if(l) q.push(l);//如果左子节点存在，则入队
                if(r) q.push(r);//如果右子节点存在，则入队
            }
        }
        return root;//返回根节点
    }

    // 传入序列化后的字符串，和当前索引
    TreeNode* getNode(string data, int& i){
        // 特判，如果第一个字符就为#,说明为空，索引移动到下一位
        if(data[i] == '#'){
            i++;
            return NULL;
        } 
        // 标记是否为负数，是否计算过数字，如果计算过，则返回一个对应节点，否则返回NULL
        bool isPositive = false, st = false;
        if(data[i] == '-'){
            i++;
            isPositive = true;
        } 
        // 遍历字符串，计算一个数字的值
        int cnt = 0;
        while(i < data.size() && data[i] != '#'){
            cnt = cnt*10 + data[i]-'0' + 0;
            i++;
            st = true;//标记计算过数字
        }
        i++;//while遍历完后索引在data[i] == #上，因此索引移动到下一位
        if(isPositive) cnt = -cnt;//如果是负数，则取相反数
        if(st) return new TreeNode(cnt);//如果计算过数字，则返回一个新的节点
        return NULL;//否则返回NULL
    }
};

// Your Codec object will be instantiated and called as such:
// Codec ser, deser;
// TreeNode* ans = deser.deserialize(ser.serialize(root));