序列化和反序列化二叉搜索树

序列化是将数据结构或对象转换为一系列位的过程,以便它可以存储在文件或内存缓冲区中,或通过网络连接链路传输,以便稍后在同一个或另一个计算机环境中重建。

设计一个算法来序列化和反序列化二叉搜索树。 确保二叉搜索树可以序列化为字符串,并且可以将该字符串反序列化为最初的二叉搜索树。

首先,定义二叉树:

from collections import deque


class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

方法一:先序遍历+BFS

def serialize(root):
    """Encodes a tree to a single string.
    :type root: TreeNode
    :rtype: str
    """
    res = []
    queue = deque()
    if root:
        queue.append(root)
    while queue:
        cur = queue.popleft()
        if cur:
            res.append(str(cur.val))
            queue.append(cur.left)
            queue.append(cur.right)
        else:
            res.append("#")
    return ",".join(res)


def deserialize(data):
    """Decodes your encoded data to tree.
    :type data: str
    :rtype: TreeNode
    """
    if not data:
        return
    data = data.split(",")[::-1]
    root = TreeNode(int(data.pop()))
    queue = deque([root])
    while queue:
        cur = queue.popleft()
        left_val = data.pop()
        if left_val != "#":
            cur.left = TreeNode(int(left_val))
            queue.append(cur.left)
        right_val = data.pop()
        if right_val != "#":
            cur.right = TreeNode(int(right_val))
            queue.append(cur.right)
    return root

方法二:后序遍历+DFS

def serialize(root):
    """Encodes a tree to a single string.
        :type root: TreeNode
        :rtype: str
    """
    if not root:
        return "#"
    return ",".join([serialize(root.left), serialize(root.right), str(root.val)])


def deserialize(data):
    """Decodes your encoded data to tree.
    :type data: str
    :rtype: TreeNode
    """
    d = data.split(",")

    def dfs():
        val = d.pop()
        if val == "#":
            return
        node = TreeNode(int(val))
        node.right = dfs()
        node.left = dfs()
        return node

    return dfs()

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/serialize-and-deserialize-bst/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值