forked from natalie-o-perret/coding-challenges
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest.py
More file actions
79 lines (64 loc) · 1.67 KB
/
test.py
File metadata and controls
79 lines (64 loc) · 1.67 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
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
class Node(object):
def __init__(self, data=None, left=None, right=None):
self.data = data
self.left = left
self.right = right
def traverse_recursive_preorder(node):
if node:
yield node.data
yield from traverse_recursive_preorder(node.left)
yield from traverse_recursive_preorder(node.right)
def traverse_iterative_preorder(node):
if node:
stack = [node]
while stack:
node = stack.pop()
yield node.data
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
def traverse_recursive_inorder(node):
if node:
yield from traverse_recursive_inorder(node.left)
yield node.data
yield from traverse_recursive_inorder(node.right)
def traverse_iterative_inorder(node):
if node:
stack = []
while stack or node:
if node:
stack.append(node)
node = node.left
else:
node = stack.pop()
yield node.data
node = node.right
def traverse_recursive_postorder(node):
if node:
yield from traverse_recursive_postorder(node.left)
yield from traverse_recursive_postorder(node.right)
yield node.data
def traverse_iterative_postorder(node):
if node:
stack = []
while stack or node:
if node:
stack.append(node)
node = node.left
else:
# if right child exists and traversing node from left child, then move right
node = stack[-1]
yield node.data
node = node.right
g = Node("G")
f = Node("F", None, g)
e = Node("E")
d = Node("D")
c = Node("C", f)
b = Node("B", d, e, )
a = Node("A", b, c)
print(*traverse_iterative_preorder(a), sep=", ")
print(*traverse_recursive_preorder(a), sep=", ")
print(*traverse_recursive_inorder(a), sep=", ")
print(*traverse_iterative_inorder(a), sep=", ")