leetcode297. 二叉树的序列化与反序列化

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Codec {
public:

    // Encodes a tree to a single string.
    string serialize(TreeNode* root) {
        stringstream ss;
        run(root, ss);
        std::cout << ss.str() << endl;
        return ss.str();
    }
    void run(TreeNode* root, stringstream& ss)
    {
        if (root)
        {
            ss << root -> val << " ";
            run(root -> left, ss);
            run(root -> right, ss);
        }
        else
        {
            ss << "# ";
        }
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        queue<string> Q;
        const char *ss = data.c_str();
        char*s = const_cast<char*> (ss);
        const char *sep = " "; 
        char *p;
        p = strtok(s, sep);
        while(p) {
            string tmp(p);
            if (!tmp.empty())
                Q.push(tmp);
            p = strtok(NULL, sep);
        }

        return build(Q);
    }
    TreeNode* build(queue<string>& Q)
    {
        if (Q.size() == 0)
            return NULL;
        string tmp = Q.front();
        Q.pop();
        if (tmp == "#")
            return NULL;
        else
        {
            TreeNode* n = new TreeNode(stoi(tmp)); 
            n -> left = build(Q);
            n -> right = build(Q); 
            return n;
        }
    }
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容