面试题37.序列化二叉树
题目
请实现两个函数,分别用来序列化和反序列化二叉树。
你需要设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
提示:输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。
思路
- 序列化:根据二叉树的先序遍历序列顺序,从根节点出发,将根节点的值加入列表中,然后递归左子树和右子树,遇到空结点,将特殊符号’#'加入列表中
- 反序列化:首先将列表放入一个迭代器中,每次取出一个字符,然后如果该字符不是’#‘,就创建一个新的节点,递归创建左子树,递归创建右子树,如果是’#',直接返回None(空结点)
代码
# 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
"""
vals = []
def encode(cur):
if not cur:
vals.append('#') # 空结点 存储特殊符号
else:
vals.append(str(cur.val))
encode(cur.left)
encode(cur.right)
encode(root)
return ' '.join(vals) # 使用空格连接字符
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
vals = iter(data.split())
def decode():
val = next(vals)
if val == '#':
return None
else:
cur = TreeNode(int(val)) # 创建一个新的节点
cur.left = decode() # 递归构建左子树
cur.right = decode() # 递归构建右子树
return cur
return decode()
# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.deserialize(codec.serialize(root))