1. 题目要求
二叉树的序列化和反序列化
2. 思路
2.1 序列化
首先我们介绍二叉树先序序列化的方式,假设序列化的结果字符串为str,初始时str等于空字符串。先序遍历二叉树,如果遇到空节点,就在str的末尾加上“#!”,“#”表示这个节点为空,节点值不存在,当然你也可以用其他的特殊字符,“!”表示一个值的结束。如果遇到不为空的节点,假设节点值为3,就在str的末尾加上“3!”。现在请你实现树的先序序列化。
2.2 反序列化
所谓反序列化是根据一个字符串重新建立一棵二叉树,反序列化是序列化的逆过程,对于一个字符串,首先按照分隔符!将其分割为字符串数组,每个字符串元素代表一个结点,然后开始重建二叉树。由于每个结点再字符串中只保留了一个val值,因此需要根据结点的值val重新构建Node结点对象,并且为这个结点对象的left和right进行赋值。
3. 代码
程序说明:
- 这里
preOrderRecur
是递归先序遍历打印二叉树。 serialByPreHelp
是二叉树序列化,返回值是string
类型。deSerialByPre
是反序列化过程。
#include <iostream>
#include <string>
struct Node {
int value;
Node* left;
Node* right;
Node(int value):
value(value), left(nullptr), right(nullptr) {}
};
std::string serialByPreHelp(Node* head) {
if (head == nullptr) {
return "#!";
}
std::string res = std::to_string(head->value) + "!";
res = res + serialByPreHelp(head->left);
res = res + serialByPreHelp(head->right);
return res;
}
//char* serialByPre(Node* head) {
// if (head == nullptr) {
// return nullptr;
// }
// std::string res = serialByPreHelp(head);
// return (char*)res.c_str();
//}
Node* deSerialByPreHelp(char*& str) {
if (*str == '#') {
str++; // skip '#'
str++; // skip '!'
return nullptr;
}
int val = 0;
while (*str != '!') { // process such '123'
val = 10 * val + (*str - '0');
str++;
}
str++;
Node* root = new Node(val); // recur
root->left = deSerialByPreHelp(str);
root->right = deSerialByPreHelp(str);
return root;
}
Node* deSerialByPre(char* str) {
if (str == nullptr || *str == '#') { // if null
return nullptr;
}
return deSerialByPreHelp(str);
}
void preOrderRecur(Node* head) {
if (head == nullptr) {
return;
}
std::cout << head->value << ",";
preOrderRecur(head->left);
preOrderRecur(head->right);
}
int main() {
// std::string res = "5!3!#!#!2!";
// std::string[] res_ = res.split("!");
Node* head = new Node(5);
head->left = new Node(3);
head->right = new Node(8);
head->left->left = new Node(2);
head->left->right = new Node(4);
head->right->left = new Node(7);
head->right->right = new Node(10);
head->right->left->left = new Node(6);
head->right->right->left = new Node(9);
head->right->right->right = new Node(11);
std::cout << "pre-order: ";
preOrderRecur(head);
std::cout << "\nserial binary: ";
std::string res = serialByPreHelp(head);
std::cout << res << std::endl;
char* s_char = (char*)res.c_str(); // convert string to char*
Node* dehead = deSerialByPre(s_char);
std::cout << "deserial binary: ";
preOrderRecur(dehead);
return 0;
}
注意
这里我们在实现序列化的过程中,我们返回值是
std::string
类型,注释的部分我们是想要返回值是char*
类型,但是这种尝试在后期调用时候失效了。我在网上找到一些是因为std::string
对象实现了拷贝构造函数和赋值构造函数了,而直接调用c_str()
方法可能会在对象的生存期失效,然后返回值就会不正确了,不过这方面确实不太懂,如果有直接将string对象直接返回为char*
方法的可以给我留言(可能借助strcpy, 全局变量?)。另外c++没有实现split
函数进行分割字符串,不像python
和java
一样,可能自己需要实现或者借助Boost C++
。