Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
serialize-and-deserialize-binary-tree.py 2.51 KB
一键复制 编辑 原始数据 按行查看 历史
Sanghee Kim 提交于 2019-01-16 03:17 +08:00 . add a solution
# Time: O(n)
# Space: O(h)
class TreeNode(object):
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Codec(object):
def serialize(self, root):
"""Encodes a tree to a single string.
:type root: TreeNode
:rtype: str
"""
def serializeHelper(node):
if not node:
vals.append('#')
return
vals.append(str(node.val))
serializeHelper(node.left)
serializeHelper(node.right)
vals = []
serializeHelper(root)
return ' '.join(vals)
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
def deserializeHelper():
val = next(vals)
if val == '#':
return None
node = TreeNode(int(val))
node.left = deserializeHelper()
node.right = deserializeHelper()
return node
def isplit(source, sep):
sepsize = len(sep)
start = 0
while True:
idx = source.find(sep, start)
if idx == -1:
yield source[start:]
return
yield source[start:idx]
start = idx + sepsize
vals = iter(isplit(data, ' '))
return deserializeHelper()
# time: O(n)
# space: O(n)
class Codec2(object):
def serialize(self, root):
"""Encodes a tree to a single string.
:type root: TreeNode
:rtype: str
"""
def gen_preorder(node):
if not node:
yield '#'
else:
yield str(node.val)
for n in gen_preorder(node.left):
yield n
for n in gen_preorder(node.right):
yield n
return ' '.join(gen_preorder(root))
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
def builder(chunk_iter):
val = next(chunk_iter)
if val == '#':
return None
node = TreeNode(int(val))
node.left = builder(chunk_iter)
node.right = builder(chunk_iter)
return node
# https://stackoverflow.com/a/42373311/568901
chunk_iter = iter(data.split())
return builder(chunk_iter)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助