Skip to content

Instantly share code, notes, and snippets.

@selimslab selimslab/codec.py
Created May 23, 2020

Embed
What would you like to do?
"""Design an algorithm to serialize and deserialize a binary tree. """
# 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
"""
def preorder(root):
vals = list()
def go(node):
if node:
vals.append(str(node.val))
go(node.left)
go(node.right)
else:
vals.append("#")
go(root)
return vals
vals = preorder(root)
return " ".join(vals)
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
def construct():
val = next(vals)
if val == "#":
return None
node = TreeNode(int(val))
node.left= construct()
node.right= construct()
return node
vals = iter(data.split())
root = construct()
return root
# codec = Codec()
# codec.deserialize(codec.serialize(root))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.