-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTree.py
More file actions
51 lines (39 loc) · 1 KB
/
Copy pathTree.py
File metadata and controls
51 lines (39 loc) · 1 KB
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
import json
def serialize_instance(obj):
return vars(obj)
class Tree:
# define the struct
def __init__(self, value):
self.value = value
self.children = []
# repr() is default method toString()
def __repr__(self):
return str(self.value)
# convert to json
def toJson(self):
return json.dumps(self, default=serialize_instance, indent=2)
def __add__(self, children):
self.children.append(children)
# implements default iter
def __iter__(self):
return iter(self.children)
# deepinly iterate first
def depth_first(self):
yield self
for i in self:
yield from i.depth_first()
root = Tree(0)
nodeF10 = Tree(10)
nodeF11 = Tree(11)
nodeF21 = Tree(21)
nodeF22 = Tree(22)
nodeF23 = Tree(23)
root.__add__(nodeF10)
root.__add__(nodeF11)
nodeF10.__add__(nodeF21)
nodeF10.__add__(nodeF22)
nodeF11.__add__(nodeF23)
print(root.toJson())
print("print by iter")
for node in root.depth_first():
print(node)