1. 程式人生 > 其它 >leetcode-297 二叉樹序列化

leetcode-297 二叉樹序列化

leetcode-297 二叉樹序列化

題目描述:

二叉樹序列化是很常用的二叉樹技巧,序列化可以唯一確定一顆二叉樹,無論是前序遍歷、中序遍歷還是後序遍歷,只要加上空指標的資訊就可以唯一確定一顆二叉樹。除此之外,將二叉樹序列化之後可以判斷兩個二叉樹是否相等,可以作為後續遍歷的返回值判斷兩個子樹是否相等。

我使用的是前序遍歷方法序列化一顆二叉樹,反序列化也採用前序遍歷的思想,因為序列化之後的第一個位置是根節點,緊隨其後的位置是他的左子樹的節點,將下一個位置接到根的左子樹,知道遇到空指標返回,右子樹同理。

root->left=deserialize(data);

最後貼程式碼:

class Codec {
public:
    string res;
    // Encodes a tree to a single string.
    string serialize(TreeNode* root) {
        serialize(root,res);
        return res;
    }
    void serialize(TreeNode* root,string s){
        if(root==NULL){
            s.append("#").append(",");
            return;
        }
        s.append(to_string(root->val)).append(",");
        serialize(root->left,s);
        serialize(root->right,s);
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        stringstream ss;
        ss << data;
        string str;
        vector<string> inputs;
        while (getline(ss, str, ','))
            inputs.emplace_back(str);
        return deserialize(inputs);
    }
    TreeNode* deserialize(vector<string>& data){
        if(data.size()==0){
            return NULL;
        }
        string temp=data[0];
        auto k = data.begin();
	    data.erase(k);
        if(temp=="#"){
            return NULL;
        }
        TreeNode* root=new TreeNode(atoi(temp.c_str()));
        root->left=deserialize(data);
        root->right=deserialize(data);
        return root;
    }
};