-
Notifications
You must be signed in to change notification settings - Fork 0
/
5.py
65 lines (50 loc) · 1.72 KB
/
5.py
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
"""
Definition of TreeNode:
class TreeNode:
def __init__(self, val):
self.val = val
self.left, self.right = None, None
"""
class Solution:
'''
@param root: An object of TreeNode, denote the root of the binary tree.
This method will be invoked first, you should design your own algorithm
to serialize a binary tree which denote by a root node to a string which
can be easily deserialized by your own "deserialize" method later.
'''
def serialize(self, root):
# write your code here
def pre_order(root):
if root:
result.append(str(root.val))
pre_order(root.left)
pre_order(root.right)
else:
result.append('#')
result = []
pre_order(root)
return '.'.join(result)
'''
@param data: A string serialized by your serialize method.
This method will be invoked second, the argument data is what exactly
you serialized at method "serialize", that means the data is not given by
system, it's given by your own serialize method. So the format of data is
designed by yourself, and deserialize it here as you serialize it in
"serialize" method.
'''
def deserialize(self, data):
# write your code here
data = data.split(',')
def change(num):
num[0] += 1
if num[0] < len(data):
if data[num[0]] == '#':
return None
root = TreeNode(data[num[0]])
root.left = change(num)
root.right = change(num)
return root
else:
return None
num = [-1]
return change(num)