Computer Science/Coding Test
LeetCode: 297. Serialize and Deserialize Binary Tree
focalpoint
2022. 1. 2. 19:08
My somewhat complicated code using queue.
Both time and memory complexities are O(N).
# 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 ''
ret = str(root.val) + ' '
q = [root]
while q:
next_q = []
for node in q:
if node.left:
next_q.append(node.left)
ret += str(node.left.val) + ' '
else:
ret += 'null '
if node.right:
next_q.append(node.right)
ret += str(node.right.val) + ' '
else:
ret += 'null '
q = next_q
return ret[:-1]
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
if not data:
return None
data = data.split(' ')
root, j = TreeNode(val=int(data[0])), 1
q = [root]
while q:
next_q = []
for node in q:
if data[j] != 'null':
node.left = TreeNode(val=int(data[j]))
next_q.append(node.left)
j += 1
if data[j] != 'null':
node.right = TreeNode(val=int(data[j]))
next_q.append(node.right)
j += 1
q = next_q
return root
# Your Codec object will be instantiated and called as such:
# ser = Codec()
# deser = Codec()
# ans = deser.deserialize(ser.serialize(root))
Similar idea but much simpler code with preorder recursive.
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
from collections import deque
class Codec:
def serialize(self, root):
"""Encodes a tree to a single string.
:type root: TreeNode
:rtype: str
"""
if not root:
return 'n'
return ','.join([str(root.val), self.serialize(root.left), self.serialize(root.right)])
def deserialize(self, data):
"""Decodes your encoded data to tree.
:type data: str
:rtype: TreeNode
"""
q = deque()
q.extend(data.split(","))
self.data = q
return self.deserialize_helper()
def deserialize_helper(self):
if self.data[0] == 'n':
self.data.popleft()
return None
node = TreeNode(self.data.popleft())
node.left = self.deserialize_helper()
node.right = self.deserialize_helper()
return node
# Your Codec object will be instantiated and called as such:
# ser = Codec()
# deser = Codec()
# ans = deser.deserialize(ser.serialize(root))