2021-5-9 剑指 Offer 37. 序列化二叉树(层序遍历 BFS)

注:
① 需要掌握 string转int,char转int,str转char的方式,C++中若需要将string类型转为int类型,需先将string转为const char,在调用std::atoi()
② 去掉string一头一尾的"[]"

 data.erase(data.begin());
 data.erase(--data.end());

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

示例:

你可以将以下二叉树:

    1
   / \
  2   3
     / \
    4   5

序列化为 “[1,2,3,null,null,4,5]”

题解:
解题思路:
通常使用的前序、中序、后序、层序遍历记录的二叉树的信息不完整,即唯一的输出序列可能对应着多种二叉树可能性。题目要求的 序列化 和 反序列化 是 可逆操作 。因此,序列化的字符串应携带 完整的二叉树信息

观察题目示例,序列化的字符串实际上是二叉树的 “层序遍历”(BFS)结果,本文也采用层序遍历。

为完整表示二叉树,考虑将叶节点下的 null 也记录。在此基础上,对于列表中任意某节点 node ,其左子节点 node.left 和右子节点 node.right 在序列中的位置都是 唯一确定 的。如下图所示:
在这里插入图片描述
序列化 Serialize :
借助队列,对二叉树做层序遍历,并将越过叶节点的 null 也打印出来。

算法流程:
特例处理: 若 root 为空,则直接返回空列表 “[]” ;
初始化: 队列 queue (包含根节点 root );序列化列表 res ;
层序遍历: 当 queue 为空时跳出;
节点出队,记为 node ;
若 node 不为空:① 打印字符串 node.val ,② 将左、右子节点加入 queue ;
否则(若 node 为空):打印字符串 “null” ;
返回值: 拼接列表,用 ‘,’ 隔开,首尾添加中括号

复杂度分析:
时间复杂度 O(N) : N 为二叉树的节点数,层序遍历需要访问所有节点,最差情况下需要访问 N+1 个 null ,总体复杂度为 O(2N + 1) = O(N) 。
空间复杂度 O(N) : 最差情况下,队列 queue 同时存储N+1个节点(或 N+1个 null ),使用 O(N) ;列表 res 使用 O(N) 。

反序列化 Deserialize :
基于本文开始推出的 node , node.left , node.right 在序列化列表中的位置关系,可实现反序列化。
利用队列按层构建二叉树,借助一个指针 i 指向节点 node 的左、右子节点,每构建一个 node 的左、右子节点,指针 i 就向右移动 11 位。

算法流程:
特例处理: 若 data 为空,直接返回 null ;
初始化: 序列化列表 vals (先去掉首尾中括号,再用逗号隔开),指针 i = 1 ,根节点 root (值为 vals[0] ),队列 queue(包含 root );
按层构建: 当 queue 为空时跳出;
节点出队,记为 node ;
构建 node 的左子节点:node.left 的值为 vals[i] ,并将 node.left 入队;
执行 i += 1 ;
构建 node 的右子节点:node.left 的值为 vals[i] ,并将 node.left 入队;
执行 i += 1 ;
返回值: 返回根节点 root 即可;

复杂度分析:
时间复杂度 O(N) : N 为二叉树的节点数,按层构建二叉树需要遍历整个 vals ,其长度最大为 2N+1 。
空间复杂度 O(N) : 最差情况下,队列 queue 同时存储N+1个节点,因此使用 O(N) 额外空间。

class Codec {
public:

    // Encodes a tree to a single string.
    string serialize(TreeNode* root) {
        string result;
        queue<TreeNode*> node;
        if(root==NULL){
            return "[]";
        }
        node.push(root);
        result+='[';
        while(!node.empty()){
            TreeNode* head=node.front();
            node.pop();
            if(head!=NULL){
                node.push(head->left);
                node.push(head->right);
                result+=(to_string(head->val)+',');
            }
            else{
                result+="null,";
            }
        }
        //删除string的最后一个元素
        result.erase(--result.end());
        result+=']';
        return result;
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        if(data=="[]"){
            return NULL;
        }
        string str;
        vector<string> result;
        //去掉string一头一尾的"[]"
        data.erase(data.begin());
        data.erase(--data.end());
        queue<TreeNode*> node;
        //得到去掉‘,’和'[]'序列化列表 result,其中每个val占用一个string
        for(int i=0;i<data.size();i++){
            if(data[i]!=','){
                str+=data[i];
            }
            else{
                result.push_back(str);
                str.clear();
            }
        }
        //C++中若需要将string类型转为int类型,需先将string转为const char*,在调用std::atoi()
        TreeNode* root=new TreeNode(atoi(result[0].c_str()));
        node.push(root);
        int i=1;
        //根据序列化列表result构建整棵树
        //i<result.size()-1,防止i溢出
        while(i<result.size()-1&&!node.empty()){
            TreeNode* head=node.front();
            node.pop();
            if(result[i]!="null"){
                head->left=new TreeNode(atoi(result[i].c_str()));
                node.push(head->left);
            }
            i++;
            if(result[i]!="null"){
                head->right=new TreeNode(atoi(result[i].c_str()));
                node.push(head->right);
            }
            i++;
        }
        return root;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值