牛客网案例较为简单,改用leetcode题目:297. 二叉树的序列化与反序列化 - 力扣(LeetCode)
题目:
序列化是将一个数据结构或者对象转换为连续的比特位的操作,进而可以将转换后的数据存储在一个文件或者内存中,同时也可以通过网络传输到另一个计算机环境,采取相反方式重构得到原数据。
请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
提示: 输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/serialize-and-deserialize-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
测试用例:
输入:[1,2,3,null,null,4,5]
输出:[1,2,3,null,null,4,5]
思路:
无论采用什么方法,关键问题是将非空结点和空结点分开。对于非空结点,因为它的值的大小、正负是不确定的,我们用一个标志 "!" 将其边界标记出来;对于空结点,可以用一个 "#" 表示。对于测试用例中的例子,如果我们使用这样的编码方式,并用前序遍历的方法遍历这棵树,可以得到字符串:1!2!#!#!3!4!#!#!5!#!#!。
再说反序列化,其实得到了这个编码后的字符串,我们就完全可以知道它的空结点的位置了。我们再使用“中左右”(和先序遍历顺序一致)的方式去构造二叉树即可。遇到 "#" 就创建一个空结点,遇到数字就创建一个TreeNode,并递归地为它的左右指针加上结点。
代码:
# 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 '#!'
s = str(root.val)+'!'
s += self.serialize(root.left)
s += self.serialize(root.right)
return s
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
def decode(data, i):
if data[i] == '#':
return None, i+1
num = 0
neg = False
if data[i] == '-':
i += 1
neg = True
while data[i] != '!':
num = num*10 + int(data[i])
i += 1
if neg:
num = -num
root = TreeNode(num)
root.left, i = decode(data, i+1)
root.right, i = decode(data, i+1)
return root, i
root, _ = decode(data, 0)
return root