leetCodePractice/LeetcodePython3/q0297.py at master · im-ling/leetCodePractice · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from collections import deque

# 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 ""
        result = ""
        bfs_queue = deque([root])
        while bfs_queue:
            p = bfs_queue.popleft()
            if p:
                result += str(p.val) + ","
                bfs_queue.append(p.left)
                bfs_queue.append(p.right)
            else:
                result += "None" + ","
        print(result[:-1])
        return result[:-1]

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

        :type data: str
        :rtype: TreeNode
        """
        if "," not in data:
            return None
        nums = data.split(",")
        length = len(nums)
        root = TreeNode(int(nums[0]))
        bfs_queue = deque([root])
        index = 1
        while bfs_queue:
            p = bfs_queue.popleft()

            if nums[index] != "None":
                p.left = TreeNode(int(nums[index]))
                bfs_queue.append(p.left)
            index += 1

            if nums[index] != "None":
                p.right = TreeNode(int(nums[index]))
                bfs_queue.append(p.right)
            index += 1
        return root

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

node2 = TreeNode(2)
node3 = TreeNode(3)
node4 = TreeNode(4)
node5 = TreeNode(5)
root = TreeNode(1)
root.left = node2
root.right = node3
node3.left = node4
node3.right = node5
result = Codec().serialize(root)
result = Codec().deserialize(result)
print(result)
print(result.val)

print("," in "asdfa asdf")