请实现两个函数,分别用来序列化和反序列化二叉树。
你需要设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
提示:输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
示例:
输入:root = [1,2,3,null,null,4,5]
输出:[1,2,3,null,null,4,5]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/xu-lie-hua-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* 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 data;
queue <TreeNode*> q;
if (root) {
q.push(root);
}
while (!q.empty()) {
auto cur = q.front();
q.pop();
if (cur) {
data += to_string(cur->val) + ",";
q.push(cur->left);
q.push(cur->right);
}
else {
data += "null,";
}
}
if (!data.empty()) {
data.pop_back();
}
return data;
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
auto dummy = new TreeNode(0);
queue <TreeNode*> q;
q.push(dummy);
int beg = 0, end = 0;
bool left_side = false;
while (beg < data.size()) {
// 截取字符串
while (end < data.size() && data[end] != ',') {
end++;
}
auto str = data.substr(beg, end - beg);
TreeNode* node = nullptr;
if (str != "null") {
// atoi是C语言中的函数,参数是字符指针char *,
// c_str()的作用是将string转换为char *
node = new TreeNode(atoi(str.c_str()));
}
auto cur = q.front();
if (left_side) {
cur->left = node;
}
else {
cur->right = node;
q.pop();
}
if (node) {
q.push(node);
}
left_side = !left_side;
beg = ++end;
}
return dummy->right;
}
};
// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));