剑指offer 37 序列化二叉树

这个题有些bug,但还是一道ok的题目。给一颗二叉树,可以将其序列化,并且还原回树。这也是一种压缩或者表示的手段。注意:叶子节点的后面跟着None。而这两种方法都可以用bfs解决。
序列化时,我们可以用bfs,如果是None,就不入队,否则入队,然后不断让儿子入队,最终完成。
解码时,我们需要遍历数组,不断地将数组赋给父节点。由于每个父亲最多两个儿子,就按顺序给就行,代码如下:

from queue import Queue
# 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 []
        l = []
        q = Queue()
        q.put(root)

        while not q.empty():
            e = q.get()
            if e:
                l.append(e.val)
                q.put(e.left)
                q.put(e.right)
            else: l.append(None)

        return l

    def deserialize(self, data):
        """Decodes your encoded data to tree.
        
        :type data: str
        :rtype: TreeNode
        """
        if data == []: return None
        root = TreeNode(data[0])
        q = Queue()
        q.put(root)
        i = 1
        while i < len(data):
            c = q.get()
            if data[i] == None: c.left = None
            else: 
                c.left = TreeNode(data[i])
                q.put(c.left)
            i += 1

            if data[i] == None: c.right = None
            else: 
                c.right = TreeNode(data[i])
                q.put(c.right)
            i += 1

        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、付费专栏及课程。

余额充值