二叉树的序列化与反序列化
序列化是将一个数据结构或者对象转换为连续的比特位的操作,进而可以将转换后的数据存储在一个文件或者内存中,同时也可以通过网络传输到另一个计算机环境,采取相反方式重构得到原数据。
请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
示例:
你可以将以下二叉树:
提示: 这与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
说明: 不要使用类的成员 / 全局 / 静态变量来存储状态,你的序列化和反序列化算法应该是无状态的。
class Codec
{
public:
// Encodes a tree to a single string.
string serialize(TreeNode *root)
{
if (!root)
{
return "[]";
}
queue<TreeNode *> Q;
string res = "[";
Q.push(root);
while (!Q.empty())
{
TreeNode *cur_node = Q.front();
Q.pop();
if (cur_node)
{
res += to_string(cur_node->val) + ",";
if (Q.size() == 0 && !cur_node->left && !cur_node->right)
{
break;
}
Q.push(cur_node->left);
Q.push(cur_node->right);
}
else
{
res += "null,";
}
}
res.pop_back();
res += "]";
return res;
}
// Decodes your encoded data to tree.
TreeNode *deserialize(string data)
{
if (data.size() == 2)
return NULL;
data.erase(data.begin());
int size = data.size();
int i = 0;
string word = "";
vector<TreeNode *> node_vec;
while (i < size)
{
while (data[i] != ',' && data[i] != ']')
{
word += data[i];
i++;
}
TreeNode *wait_to_push = (word == "null") ? NULL : new TreeNode(stoi(word));
node_vec.push_back(wait_to_push);
word = "";
i++;
}
int j = 0;
queue<TreeNode *> Q;
Q.push(node_vec[j++]);
while (!Q.empty())
{
TreeNode *cur_node = Q.front();
Q.pop();
if (cur_node)
{
if (j < node_vec.size())
{
TreeNode *node1 = node_vec[j++];
cur_node->left = node1;
// cout << cur_node->val << " left is " << node1->val << endl;
Q.push(node1);
}
if (j < node_vec.size())
{
TreeNode *node2 = node_vec[j++];
cur_node->right = node2;
// cout << cur_node->val << " right is " << node2->val << endl;
Q.push(node2);
}
}
}
// cout<<node_vec[0]->right->right->val<<endl;
return node_vec[0];
}
};