37. 序列化二叉树
请实现两个函数,分别用来序列化和反序列化二叉树。
你需要设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
提示:输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-OOd9Q0bH-1651224708468)(C:\Users\陈术山\AppData\Roaming\Typora\typora-user-images\image-20220429083729976.png)]
示例:
输入:root = [1,2,3,null,null,4,5]
输出:[1,2,3,null,null,4,5] //先序遍历
解题思路
深度优先搜索(先序遍历)
class Codec {
public:
// Encodes a tree to a single string.
string serialize(TreeNode* root) {
string res;
rserialize(root, res);
return res;
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
list<string> dataArray;
string str;
for(auto& ch:data) {
if(ch == ',') {
dataArray.push_back(str);
str.clear();
} else {
str.push_back(ch);
}
}
if(!str.empty()) {
dataArray.push_back(str);
str.clear();
}
return rdeserialize(dataArray);
}
private:
void rserialize(TreeNode* root, string& res) {
if(root == nullptr) {
res += "None,";
} else {
res += to_string(root->val) + ",";
rserialize(root->left, res);
rserialize(root->right, res);
}
}
TreeNode* rdeserialize(list<string>& dataArray) {
if(dataArray.front() == "None") {
dataArray.erase(dataArray.begin());
return nullptr;
}
TreeNode* root = new TreeNode(stoi(dataArray.front()));
dataArray.erase(dataArray.begin());
root->left = rdeserialize(dataArray);
root->right = rdeserialize(dataArray);
return root;
}
};