剑指offer:序列化二叉树
admin
2023-07-07 06:44:15
0

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

# -*- coding: utf-8 -*-
# @Time         : 2019-07-07 15:48
# @Author       : Jayce Wong
# @ProjectName  : job
# @FileName     : serializeBinaryTree.py
# @Blog         : https://blog.51cto.com/jayce1111
# @Github       : https://github.com/SysuJayce

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

class Solution:
    """
    用先序遍历将二叉树序列化下来,然后再反序列化即可。因为这里的关键在于如何定位到根节点,虽然用后
    序遍历也可以定位根节点,但是在反序列化的时候就不容易实现。
    """
    def Serialize(self, root):
        """
        序列化的时候正常先序遍历二叉树即可,但是为了准确定位到叶子节点,我们需要
        **将空节点也序列化下来**
        """
        def helper(pRoot, curSerial):
            if not pRoot:
                # 这里我们用'$'表示空节点
                curSerial.append('$')
                return

            # 用递归方法进行序列化,先将根节点序列化,然后遍历左子树、右子树
            curSerial.append(pRoot.val)
            helper(pRoot.left, curSerial)
            helper(pRoot.right, curSerial)

        if not root:
            return ''

        serialization = []
        helper(root, serialization)
        return ','.join(map(str, serialization))

    def Deserialize(self, s):
        """
        在反序列化的时候,由于构造一个节点的时候默认左右子节点都是空,因此如果字符串中遇到了'$',
        可以直接忽略。也就是只需要处理非空节点
        """
        def helper(curStr, curRoot):
            # 递归出口
            if not curStr:
                return curRoot
            # 这里由于我们使用一个列表保存节点信息,因此可以用pop()来保存剩余待反序列化的节点
            curVal = curStr.pop(0)
            # 如果是'$',即该节点为空,那么这时候由于输入的curRoot也为空,直接返回即可
            if curVal != '$':
                # 从整体来看,我们需要先构造根节点,然后分别构造其左右子节点。
                # 递归在确定了出口条件之后,就只需要将整体逻辑写出来就行了。
                curRoot = TreeNode(int(curVal))
                curRoot.left = helper(curStr, curRoot.left)
                curRoot.right = helper(curStr, curRoot.right)

            return curRoot

        if not s:
            return None
        s = s.split(',')
        root = helper(s, None)
        return root

def printTree(root: TreeNode):
    if not root:
        return
    print(root.val)
    printTree(root.left)
    printTree(root.right)

def main():

    solution = Solution()
    root = TreeNode(10)
    root.left = TreeNode(6)
    root.right = TreeNode(14)
    root.left.left = TreeNode(4)
    root.left.right = TreeNode(8)
    root.right.left = TreeNode(12)
    root.right.right = TreeNode(16)
    s = solution.Serialize(root)
    print(s)
    printTree(solution.Deserialize(s))

if __name__ == '__main__':
    main()

相关内容

热门资讯

我国科学家为细胞信号“导航”开... 新华社济南5月31日电(记者张力元)人体细胞犹如一座精密的通信城市,每天都有大量“指令”穿梭传递,调...
极端大风突袭哈尔滨!过山车停摆... 极目新闻记者 詹钘5月31日,受强对流天气影响,哈尔滨国际会展中心体育场相关设施受到损坏,原计划当晚...
三原电缆取得电缆接头连接用防护... 国家知识产权局信息显示,上海三原电缆附件有限公司取得一项名为“一种电缆接头连接用防护结构”的专利,授...
原创 识... 还是那句话,机圈苦大屏久已…… 虽然大屏有大屏的美,但是小屏也有小屏的俏。在大屏旗舰占据主流的手机市...
玄戒技术取得分频电路专利,实现... 国家知识产权局信息显示,北京玄戒技术有限公司取得一项名为“分频电路、分频器、射频芯片和电子设备”的专...
为什么今年香会基调明显变了 5月29日—31日在新加坡举行的第23届香格里拉对话会(简称“香会”),见证着元首引领下大国关系继续...
成本几毛钱、假驱蚊液香精兑水,... 入夏升温,蚊虫进入活跃期,驱蚊防护成为民生刚需,《财经调查》持续接到消费者投诉,他们买到的多款网红驱...
越来越多80后90后,正在丧失... 六一儿童节到来之际,朋友圈里开始出现一种熟悉的热闹。有人晒出零食礼包,有人半开玩笑地向伴侣讨礼物,还...
洋保电子取得用于低温环境的电气... 国家知识产权局信息显示,洋保电子(太仓)有限公司取得一项名为“一种用于低温环境的电气柜”的专利,授权...
中日韩飞手争霸宁波!2026无... 潮新闻客户端 记者 陈冲 通讯员 朱凝 5月31日,2026小遛·无人机竞速世界杯(中国·宁波鄞州站...