剑指 Offer -- 序列化二叉树(六十一)

序列化二叉树(六十一)

题目描述:
请实现两个函数，分别用来序列化和反序列化二叉树

代码(已在牛客 AC)

这道题呢, 可以使用二叉树的前序遍历将二叉树进行序列化, 由于用的是 C++, 因此使用 <sstream> 中定义的 ostringstream 和 istringstream 还是非常方便的.

使用 serialize(root, out) 函数对二叉树进行前序遍历, 如果遇到空节点, 那么输出 # 后直接返回; 否则输出 root->val. 之后要将 ostringstream 中的字符拷出来, 应该使用 out.str() 返回保存的字符串, 而要后的 C 风格的字符串, 需要调用 c_str() 方法.

对于反序列化, 使用 deserialize 方法实现, 其中还使用了 istringstream 对象, 使用前序遍历, 对于输入到 val 中的值, 判断它是否为 #, 如果是的话, 那么说明是空节点, 返回 nullptr, 否则返回 TreeNode(stoi(val)).

class Solution {
public:
    char* Serialize(TreeNode *root) {    
        ostringstream out;
        serialize(root, out);
        char *res = new char[out.str().length() + 1];
        strcpy(res, out.str().c_str());
        return res;
    }
    TreeNode* Deserialize(char *str) {
        istringstream in(str);
        return deserialize(in);
    }
private:
    // 不论是序列化还是反序列化, 都使用二叉树的前序遍历.
    void serialize(TreeNode *root, ostringstream &out) {
        if (!root) { out << "# "; return; }
        out << root->val << " ";
        serialize(root->left, out);
        serialize(root->right, out);
    }

    TreeNode* deserialize(istringstream &in) {
        string val;
        in >> val;
        if (val == "#") return nullptr;
        TreeNode *root = new TreeNode(stoi(val));
        root->left = deserialize(in);
        root->right = deserialize(in);
        return root;
    }
};