|
| 1 | +# Serialize and Deserialize Binary Tree |
| 2 | +# https://leetcode.com/problems/serialize-and-deserialize-binary-tree/ |
| 3 | + |
| 4 | +# Definition for a binary tree node. |
| 5 | +# class TreeNode(object): |
| 6 | +# def __init__(self, x): |
| 7 | +# self.val = x |
| 8 | +# self.left = None |
| 9 | +# self.right = None |
| 10 | + |
| 11 | +class Codec: |
| 12 | + |
| 13 | + def preOrder(self, node, array): |
| 14 | + if not node: |
| 15 | + array.append("null") |
| 16 | + return |
| 17 | + array.append(str(node.val)) |
| 18 | + self.preOrder(node.left, array) |
| 19 | + self.preOrder(node.right, array) |
| 20 | + |
| 21 | + def serialize(self, root): |
| 22 | + array = [] |
| 23 | + self.preOrder(root, array) |
| 24 | + # Join the result using comma as delimiter. |
| 25 | + return ",".join(array) |
| 26 | + """Encodes a tree to a single string. |
| 27 | + :type root: TreeNode |
| 28 | + :rtype: str |
| 29 | + """ |
| 30 | + def createNode(self, val): |
| 31 | + return TreeNode(int(val)) |
| 32 | + |
| 33 | + def preOrderCreateTree(self, array, index): |
| 34 | + # The reason why we use index as list and not a single number is because a single number gets modified within recurssive calls but a list does not. |
| 35 | + if (index[0] > len(array) or array[index[0]] == 'null'): |
| 36 | + return |
| 37 | + # Create the node with the value. |
| 38 | + node = self.createNode(array[index[0]]) |
| 39 | + # Increment the index and get the left side of the node. |
| 40 | + index[0] = index[0] + 1 |
| 41 | + node.left = self.preOrderCreateTree(array, index) |
| 42 | + # Increment the index and get the right side of the node. |
| 43 | + index[0] = index[0] + 1 |
| 44 | + node.right = self.preOrderCreateTree(array, index) |
| 45 | + # Return this node, so it can be added to left or right of the calling parent. |
| 46 | + return node |
| 47 | + |
| 48 | + def deserialize(self, data): |
| 49 | + if not data: |
| 50 | + return [] |
| 51 | + # Split the data into a list, using comma as delimiter. |
| 52 | + nodeList = data.split(",") |
| 53 | + # Create the tree. |
| 54 | + node = self.preOrderCreateTree(nodeList, [0]) |
| 55 | + # Return the root of the tree. |
| 56 | + return node |
| 57 | + """Decodes your encoded data to tree. |
| 58 | + :type data: str |
| 59 | + :rtype: TreeNode |
| 60 | + """ |
| 61 | + |
| 62 | + |
| 63 | +# Your Codec object will be instantiated and called as such: |
| 64 | +# ser = Codec() |
| 65 | +# deser = Codec() |
| 66 | +# ans = deser.deserialize(ser.serialize(root)) |
0 commit comments