Leetcode 297 Solution

This article provides solution to leetcode question 297 (serialize-and-deserialize-binary-tree)
Categories:
Tags:
Access this page by simply typing in "lcs 297" in your browser address bar if you have bunnylol configured.

https://leetcode.com/problems/serialize-and-deserialize-binary-tree

Solution

# 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
        """



        tokens = []



        def _serialize(node):
            nonlocal tokens



            if node == None:
                tokens.append("#")
                return



            tokens.append(str(node.val))
            _serialize(node.left)
            _serialize(node.right)



        _serialize(root)



        return " ".join(tokens)



    def deserialize(self, data):
        """Decodes your encoded data to tree.



        :type data: str
        :rtype: TreeNode
        """
        tokens = data.split(" ")



        i = 0
        def _deserialize():
            nonlocal i



            if i == len(tokens):
                return None



            if tokens[i] == '#':
                i += 1
                return None



            node = TreeNode(int(tokens[i]))
            i += 1



            node.left = _deserialize()
            node.right = _deserialize()



            return node



        return _deserialize()



# Your Codec object will be instantiated and called as such:
# ser = Codec()
# deser = Codec()
# ans = deser.deserialize(ser.serialize(root))