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

题目描述

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

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

 

示例

你可以将以下二叉树：

    1
   / \
  2   3
     / \
    4   5

序列化为 "[1,2,3,null,null,4,5]"
提示: 这与 LeetCode 目前使用的方式一致，详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式，你也可以采用其他的方法解决这个问题。

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

 

思路

序列化可以用前序遍历。

而反序列化因为前序遍历包含了空节点，所以只根据前序遍历结果可以得到唯一得树。

 

代码

/**
 * 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) {
        string res;
        dfs1(root, res);
        return res;
    }
    
    void dfs1(TreeNode* root, string& res){
        if (!root){
            res += "null,";
            return;
        }
        res += to_string(root->val) + ",";
        dfs1(root->left, res);
        dfs1(root->right, res);
    }
    
    TreeNode* dfs2(string& data, int& n){
        if (data[n] == 'n'){
            n += 5;
            return nullptr;
        }
        bool bMinus = false;
        int val = 0;
        if (data[n] == '-'){
            bMinus = true;
            n++;
        }
        while (data[n] >= '0' && data[n] <= '9'){
            val = val * 10 + data[n] - '0';
            n++;
        }
        n++;
        if (bMinus) val *= -1;
        TreeNode* root = new TreeNode(val);
        root->left = dfs2(data, n);
        root->right = dfs2(data, n);
        return root;
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        int n = 0;
        return dfs2(data, n);
    }
};