剑指offer - 61序列化二叉树

题目描述

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

思路如下：

1. 使用一个字符串保存二叉树序列化的结果，序列化和反序列化，就是分别实现 二叉树转字符串，字符串转二叉树

2. 序列化时，使用前序递归遍历， 将节点的值转为字符串，后跟一个逗号，如果是空节点，则用#代替

3. 反序列化时，将序列的过程反过来即可

代码如下：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    //使用一个字符串保存二叉树序列化的结果
    string treestr;
    //采用递归的方法，前序遍历二叉树， 
    //每个节点的值转为字符串，后面都跟一个逗号，如果节点为空，则使用#代替
    char* Serialize(TreeNode *root) {    
        stringstream ss;
        if(NULL != root){
            ss << root->val;
            treestr += ss.str();
            treestr += ',';
            if(NULL != root->left){
                Serialize(root->left);
            }
            else{
                treestr += "#,";
            }
            if(NULL != root->right){
                Serialize(root->right);
            }
            else{
                treestr += "#,";
            }
        }
        else{
            treestr += "#,";
        }
        return (char*)treestr.c_str();
    }
    
    TreeNode* decode(char*& str){
        if(*str == '#'){  //遇到# 则跳过#,两个字符，说明此节点是空
            str += 2;
            return NULL;
        }
        int num = 0;
        while(*str != ','){       //字符不是逗号 则计算数值
            num = num * 10 + (*str - '0');
            str++;
        }
        str++;
        TreeNode* node = new TreeNode(num);
        node->left  = decode(str);
        node->right = decode(str);
        return node;
    }
    
    TreeNode* Deserialize(char *str) {
        return decode(str);
    }
};