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

题目描述

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

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

示例: 

你可以将以下二叉树：

    1
   / \
  2   3
     / \
    4   5

序列化为 "[1,2,3,null,null,4,5]"

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

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

问题分析

此题我选用例子中的层序遍历方式进行序列化与反序列化。序列化时创建一个ostringstream类型的变量out，然后在层序遍历过程中，遇到非空节点则将节点的值与空格存入out，然后将它的左右孩子加入队列；遇到空节点则将字符串“null”与空格存入out。层序遍历结束后，返回out.str()，这个函数将刚才存入out的各种类型的变量都转为了字符串并连接起来，完成了序列化。反序列化时还是层序遍历，创建一个istringstream类型的变量in，每次in >> s都会将一个字符串保存到s，是以空格自动分隔的。先创建根节点root，然后将其加入队列，然后开始层序遍历，进行in >> s，当s为空时，跳出循环，然后判断当s不为“null”时，以s的值创建左孩子，然后将左孩子加入队列，然后in >> s，当s为空时，跳出循环，然后判断当s不为“null”时，以s的值创建右孩子，然后将右孩子加入队列。当循环结束后，返回root。

代码实现

/**
 * 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) {
        ostringstream out;
        queue<TreeNode*> q;
        if(root)
            q.push(root);
        while(!q.empty()){
            TreeNode* temp = q.front();
            q.pop();
            if(temp){
                out << temp->val << " ";
                q.push(temp->left);
                q.push(temp->right);
            }
            else
                out << "null ";
        }
        return out.str();
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        istringstream in(data);
        queue<TreeNode*> q;
        string s;
        if(!(in >> s))
            return NULL;
        TreeNode* root = new TreeNode(stoi(s));
        q.push(root);
        while(!q.empty()){
            TreeNode* temp = q.front();
            q.pop();
            if(!(in >> s))
                break;
            if(s != "null"){
                temp->left = new TreeNode(stoi(s));
                q.push(temp->left);
            }
            if(!(in >> s))
                break;
            if(s != "null"){
                temp->right = new TreeNode(stoi(s));
                q.push(temp->right);
            }
        }
        return root;
    }
};

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