剑指offer树37.序列化二叉树

Posted 2021-02-28 trevo

递归

前序遍历序列化成字符串，在反序列化回去

/**
 * 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) {
        string res;
        dfs_s(root, res);
        return res;
    }
    void dfs_s(TreeNode* root, string &res)
    {
        if(!root) {
            res += "null ";
            return ;
        }
        res += to_string(root->val) + ‘ ‘;
        dfs_s(root -> left, res);
        dfs_s(root -> right, res);
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        int u = 0;
        return dfs_d(data, u);
    }

    TreeNode* dfs_d(string data, int &u)
    {
        if(u == data.size()) return NULL;
        int k = u;
        while(data[k] != ‘ ‘) k++;
        if(data[u] == ‘n‘){
            u = k + 1;
            return NULL;
        }
        int val = 0;
        if(data[u] == ‘-‘){
            for (int i = u+1; i < k; i++) val = val * 10 + data[i] - ‘0‘;
            val  = -val;
        }
        else{
        //如果是数字是正的
            for (int i = u; i < k; i++) val = val * 10 + data[i] - ‘0‘;
        }
        u = k + 1;
        auto root = new TreeNode(val);
        root -> left = dfs_d(data, u);
        root -> right = dfs_d(data, u);
        return root;
    }
};

// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));