序列化二叉树

最新推荐文章于 2024-10-15 12:30:27 发布

qq_43592587

最新推荐文章于 2024-10-15 12:30:27 发布

阅读量60

点赞数

分类专栏：剑指offer 文章标签：算法 python

本文链接：https://blog.csdn.net/qq_43592587/article/details/104688683

版权

剑指offer 专栏收录该内容

36 篇文章 0 订阅

订阅专栏

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

二叉树的序列化是指：把一棵二叉树按照某种遍历方式的结果以某种格式保存为字符串，从而使得内存中建立起来的二叉树可以持久保存。序列化可以基于先序、中序、后序、层序的二叉树遍历方式来进行修改，序列化的结果是一个字符串，序列化时通过某种符号表示空节点（#），以！表示一个结点值的结束（value!）。

二叉树的反序列化是指：根据某种遍历顺序得到的序列化字符串结果str，重构二叉树。

思路一：
层序遍历

代码：

class Solution:
    def Serialize(self, root):
        # write code here
        queue = []
        queue.append(root)
        s=''
        while queue:
            p = queue.pop(0)
            if p:
                queue.append(p.left)
                queue.append(p.right)
                s += str(p.val)
            else: s += '#'
            s += ','
        s += '!'
        return s
    def Deserialize(self, s):
        # write code here
        if len(s) < 4: return None
        arr = s.split(',')
        root = TreeNode(int(arr[0]))
        queue = [root]
        i = 1
        while queue:
            p = queue.pop(0)
            if arr[i] != '#' and i < len(arr)-1:
                p.left = TreeNode(int(arr[i]))
                queue.append(p.left)
            i += 1
            if arr[i] != '#' and i < len(arr)-1:
                p.right = TreeNode(int(arr[i]))
                queue.append(p.right)
            i += 1
        return root

思路二：
前序遍历

代码：

class Solution:
    def __init__(self):
        self.i = 0
    def Serialize(self, root):
        # write code here
        def helper(root):
            if not root: return '#,'
            s = ''
            s = s + str(root.val) + ','
            s = s + helper(root.left)
            s = s + helper(root.right)
            return s
        s = helper(root)
        s += '!'
        return s
    def Deserialize(self, s):
        # write code here
        def helper(arr):
            if arr[self.i] == '#' or self.i > len(arr)-2: return None
            root = TreeNode(int(arr[self.i]))
            self.i += 1
            root.left = helper(arr)
            self.i += 1
            root.right = helper(arr)
            return root
        arr = s.split(',')
        root = helper(arr)
        return root