C++实现二叉树的序列化和反序列化

1. 题目要求

二叉树的序列化和反序列化

2. 思路

2.1 序列化

首先我们介绍二叉树先序序列化的方式，假设序列化的结果字符串为str，初始时str等于空字符串。先序遍历二叉树，如果遇到空节点，就在str的末尾加上“#!”，“#”表示这个节点为空，节点值不存在，当然你也可以用其他的特殊字符，“!”表示一个值的结束。如果遇到不为空的节点，假设节点值为3，就在str的末尾加上“3!”。现在请你实现树的先序序列化。

2.2 反序列化

所谓反序列化是根据一个字符串重新建立一棵二叉树，反序列化是序列化的逆过程，对于一个字符串，首先按照分隔符！将其分割为字符串数组，每个字符串元素代表一个结点，然后开始重建二叉树。由于每个结点再字符串中只保留了一个val值，因此需要根据结点的值val重新构建Node结点对象，并且为这个结点对象的left和right进行赋值。

3. 代码

程序说明：

1. 这里preOrderRecur是递归先序遍历打印二叉树。
2. serialByPreHelp是二叉树序列化，返回值是string类型。
3. 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++。

4. 参考文献

1. 二叉树的序列化和反序列化实现(C++)
2. 二叉树4：二叉树的序列化和反序列化(Java版本)
3. C++返回值为对象时复制构造函数不执行怎么破
4. 知乎：C++ 的 string 为什么不提供 split 函数？
5. char * 与 string 类型相互转换方法–Linux C/C++