LeetCode解题心得——二叉树的序列化与反序列化(python)

题目

序列化是将一个数据结构或者对象转换为连续的比特位的操作,进而可以将转换后的数据存储在一个文件或者内存中,同时也可以通过网络传输到另一个计算机环境,采取相反方式重构得到原数据。

请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。

示例:

你可以将以下二叉树:

  1
 / \
2   3
    / \
   4   5

序列化为 “[1,2,3,null,null,4,5]”
提示: 这与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。

说明: 不要使用类的成员 / 全局 / 静态变量来存储状态,你的序列化和反序列化算法应该是无状态的。

思路

时间复杂度O(n),空间复杂度O(n)

序列化使用BFS,将遍历结果添加到列表返回
反序列化时,需要一个队列来保存当前根的左右子节点,每到一个新的节点,判断其是否有子节点,有的话就加入队列,从队列取出一个节点作为当前根节点

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Codec:

    def serialize(self, root):
        """Encodes a tree to a single string.
        
        :type root: TreeNode
        :rtype: str
        """
        if not root: return []
        queue = [root]
        ans = []
        while queue:
            cur = queue.pop(0)
            if cur == None:
                ans.append('null')
                continue
            ans.append(cur.val)
            queue.append(cur.left)
            queue.append(cur.right)
        return ans
                
    def deserialize(self, data):
        """Decodes your encoded data to tree.
        
        :type data: str
        :rtype: TreeNode
        """
        if not data: return
        root = TreeNode(data.pop(0))
        ROOT = root
        stack = []
        while data:
            val = data.pop(0)
            if val != 'null':
                root.left = TreeNode(val)
            if root.left and root.left.val != 'null': stack.append(root.left)
            val = data.pop(0)
            if val != 'null':
                root.right = TreeNode(val)
            if root.right and root.right.val != 'null': stack.append(root.right)
            if stack: root = stack.pop(0)
        return ROOT
# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.deserialize(codec.serialize(root))
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值