297. 二叉树的序列化与反序列化
原始题目链接:https://leetcode.cn/problems/serialize-and-deserialize-binary-tree/
序列化是将一个数据结构或者对象转换为连续的比特位的操作,进而可以将转换后的数据存储在一个文件或者内存中,同时也可以通过网络传输到另一个计算机环境,采取相反方式重构得到原数据。
请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
提示: 输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
解题思路:
使用BFS或者DFS来序列化和反序列化二叉树。使用BFS借助双端队列数据结构,来记录遍历到哪个节点了。DFS就是使用递归的方式。
代码实现:
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
"""
if not root:
return ""
# 序列化操作:双端队列模拟BSF遍历
queue = collections.deque([root])
# 存储序列化的节点,最后转成字符串形式
res = []
while queue:
# BSF遍历到的当前节点在双端队列的队头
# 队列从左侧出队表示队头元素开始序列化
node = queue.popleft()
if node:
res.append(str(node.val))
# BFS遍历当前节点的左右子节点入队
queue.append(node.left)
queue.append(node.right)
else:
# 当前节点为空,插入一个空值占位
res.append("None")
return '[' + ','.join(res) + ']'
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
# 反序列化操作
if not data:
return []
# 从1到-1,过滤掉data中的'['和']'字符
data_list = data[1:-1].split(',')
# 根节点就是BFS遍历的第一个节点
root = TreeNode(int(data_list[0]))
# 使用双端队列辅助构建二叉树,记录遍历到哪个节点了
queue = collections.deque([root])
# 从根节点的下一个节点开始,所以i=1
i = 1
while queue:
node = queue.popleft()
# 注意是与字符‘None’判断
if data_list[i] != 'None':
node.left = TreeNode(int(data_list[i]))
# 当前节点入队,用于继续构建二叉树
queue.append(node.left)
i += 1
if data_list[i] != 'None':
node.right = TreeNode(int(data_list[i]))
queue.append(node.right)
i += 1
return root
# Your Codec object will be instantiated and called as such:
# ser = Codec()
# deser = Codec()
# ans = deser.deserialize(ser.serialize(root))
DFS方式
# 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 "None"
return str(root.val) + ',' + str(self.serialize(root.left)) + ',' + str(self.serialize(root.right))
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
def dfs(data_list):
val = data_list.pop(0)
if val == "None":
return None
root = TreeNode(int(val))
root.left = dfs(data_list)
root.right = dfs(data_list)
return root
data_list = data.split(',')
return dfs(data_list)
# Your Codec object will be instantiated and called as such:
# ser = Codec()
# deser = Codec()
# ans = deser.deserialize(ser.serialize(root))
参考文献:
https://leetcode.cn/problems/serialize-and-deserialize-binary-tree/solution/297-er-cha-shu-de-xu-lie-hua-yu-fan-xu-l-647c/