C++ 二叉查找树的编码与解码

序列化和反序列化二叉搜索树

题目[leetcode449]

序列化是将数据结构或对象转换为一系列位的过程，以便它可以存储在文件或内存缓冲区中，或通过网络连接链路传输，以便稍后在同一个或另一个计算机环境中重建。

设计一个算法来序列化和反序列化二叉搜索树。 对序列化/反序列化算法的工作方式没有限制。 您只需确保二叉搜索树可以序列化为字符串，并且可以将该字符串反序列化为最初的二叉搜索树。

编码的字符串应尽可能紧凑。

注意：不要使用类成员/全局/静态变量来存储状态。 你的序列化和反序列化算法应该是无状态的。

思路

①二叉查找树编码为字符串：
将二叉查找树前序遍历，遍历时将整型的数据转为字符串，并将这些字符串数据进行连接，连接时使用特殊符号分隔。

如何插入节点，点击：二叉查找树的基础知识 了解

②字符串解码为二叉查找树：
将字符串按照编码时的分隔符“#”，将各个数字逐个拆解出来，再按照前序遍历的顺序插入到二叉查找树的根节点中，最后返回根节点，即完成解码。

实现代码如下：

class Codec {
public:

    // Encodes a tree to a single string.
    string serialize(TreeNode* root) {
        string data;
        BST_preorder(root, data);
        return data;
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        if (data.length() == 0){
	    	return NULL;
	    }
    	std::vector<TreeNode *> node_vec;
    	int val = 0;
    	for (int i = 0; i < data.length(); i++){
	    	if (data[i] == '#'){
	    		node_vec.push_back(new TreeNode(val));
	    		val = 0;
	    	}
	    	else{
	    		val = val * 10 + data[i] - '0';
	    	}
	    }
	    for (int i = 1; i < node_vec.size(); i++){
    		BST_insert(node_vec[0], node_vec[i]); //插入节点
    	}
    	return node_vec[0];
    }
private:
    void BST_insert(TreeNode *node, TreeNode *insert_node){
        if (insert_node->val < node->val){
            if (node->left){
                BST_insert(node->left, insert_node);
            }
            else{
                node->left = insert_node;
            }
        }
        else{
            if (node->right){
                BST_insert(node->right, insert_node);
            }
            else{
                node->right = insert_node;
            }
        }
    }

    void change_int_to_string(int val, std::string &str_val){
        std::string tmp;
        while(val){
            tmp += val % 10 + '0'; //在尾部加上字符
            val = val / 10;
        }
        for (int i = tmp.length() - 1; i >= 0; i--){
            str_val += tmp[i]; //逆序加入到字符串中
        }
        str_val += '#';
    }

    void BST_preorder(TreeNode *node, std::string &data){
        if (!node){
            return;
        }
        std::string str_val;
        change_int_to_string(node->val, str_val);
        data += str_val;
        BST_preorder(node->left, data);
        BST_preorder(node->right, data);
    }
};

致谢

本章知识点和思路由小象学院相关视频提供，由本人学习并梳理得出，希望自己加深记忆的同时，也能给大家提供更多有关于一些算法的知识点。
你的点赞、评论、收藏就是对我最大的支持与鼓励，谢谢！