297. 二叉树的序列化与反序列化
原始题目链接:https://leetcode-cn.com/problems/serialize-and-deserialize-binary-tree/
序列化是将一个数据结构或者对象转换为连续的比特位的操作,进而可以将转换后的数据存储在一个文件或者内存中,同时也可以通过网络传输到另一个计算机环境,采取相反方式重构得到原数据。
请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
提示: 输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
输入:root = [1,2,3,null,null,4,5]
输出:[1,2,3,null,null,4,5]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[1,2]
解题思路:
数据结构借助双端队列实现,根据输入输出要求,序列化二叉树参考广度优先搜索(BFS)来实现,反序列化二叉树根据列表中的元素构建树中的节点,注意构建顺序即可。
代码实现:
# 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
"""
# 使用队列操作序列化二叉树,数据结构使用collections中的双端队列(参考BFS优先搜索)
if not root:
return "[]"
queue = collections.deque()
# 将根节点入队
queue.append(root)
# 用于存储序列化结果的列表
res = []
while queue:
node = queue.popleft()
if node:
res.append(str(node.val))
# 将当前节点的左右子节点入队
queue.append(node.left)
queue.append(node.right)
else:
res.append("null")
return '[' + ','.join(res) + ']'
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
# 反序列化二叉树,数据结构同样可以使用双端队列进行操作
if data == "[]":
return
# 1和-1的切片索引用来过滤掉'[' ']'两个字符,
# i可看做指针或者偏移量计数器,初始化为1,即elements数组中的第二个节点
elements, i = data[1:-1].split(','), 1
# 构建根节点
root = TreeNode(int(elements[0]))
queue = collections.deque()
queue.append(root)
while queue:
# 队列中的元素出队,用于构建该元素的左右子树
node = queue.popleft()
# 构建左子树
if elements[i] != "null":
node.left = TreeNode(int(elements[i]))
queue.append(node.left)
# 偏移量+1,指向下一个元素
i += 1
# 构建右子树
if elements[i] != "null":
node.right = TreeNode(int(elements[i]))
queue.append(node.right)
# 偏移量+1,指向下一个元素
i += 1
return root
# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.deserialize(codec.serialize(root))
参考文献:
https://leetcode-cn.com/problems/xu-lie-hua-er-cha-shu-lcof/solution/mian-shi-ti-37-xu-lie-hua-er-cha-shu-ceng-xu-bian-/