1
+ # flake8: noqa
2
+
1
3
"""
2
4
This is pure Python implementation of tree traversal algorithms
3
5
"""
@@ -50,8 +52,8 @@ def pre_order(node: TreeNode) -> None:
50
52
>>> root.left, root.right = tree_node2, tree_node3
51
53
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
52
54
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
53
- >>> pre_order(root).strip()
54
- 1 2 4 5 3 6 7
55
+ >>> pre_order(root)
56
+ 1 2 4 5 3 6 7
55
57
"""
56
58
if not isinstance (node , TreeNode ) or not node :
57
59
return
@@ -72,8 +74,8 @@ def in_order(node: TreeNode) -> None:
72
74
>>> root.left, root.right = tree_node2, tree_node3
73
75
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
74
76
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
75
- >>> in_order(root).strip()
76
- 4 2 5 1 6 3 7
77
+ >>> in_order(root)
78
+ 4 2 5 1 6 3 7
77
79
"""
78
80
if not isinstance (node , TreeNode ) or not node :
79
81
return
@@ -94,8 +96,8 @@ def post_order(node: TreeNode) -> None:
94
96
>>> root.left, root.right = tree_node2, tree_node3
95
97
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
96
98
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
97
- >>> post_order(root).strip()
98
- 4 5 2 6 7 3 1
99
+ >>> post_order(root)
100
+ 4 5 2 6 7 3 1
99
101
"""
100
102
if not isinstance (node , TreeNode ) or not node :
101
103
return
@@ -116,8 +118,8 @@ def level_order(node: TreeNode) -> None:
116
118
>>> root.left, root.right = tree_node2, tree_node3
117
119
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
118
120
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
119
- >>> level_order(root).strip()
120
- 1 2 3 4 5 6 7
121
+ >>> level_order(root)
122
+ 1 2 3 4 5 6 7
121
123
"""
122
124
if not isinstance (node , TreeNode ) or not node :
123
125
return
@@ -144,10 +146,10 @@ def level_order_actual(node: TreeNode) -> None:
144
146
>>> root.left, root.right = tree_node2, tree_node3
145
147
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
146
148
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
147
- >>> level_order_actual(root).strip()
148
- 1
149
- 2 3
150
- 4 5 6 7
149
+ >>> level_order_actual(root)
150
+ 1
151
+ 2 3
152
+ 4 5 6 7
151
153
"""
152
154
if not isinstance (node , TreeNode ) or not node :
153
155
return
@@ -180,8 +182,8 @@ def pre_order_iter(node: TreeNode) -> None:
180
182
>>> root.left, root.right = tree_node2, tree_node3
181
183
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
182
184
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
183
- >>> pre_order_iter(root).strip()
184
- 1 2 4 5 3 6 7
185
+ >>> pre_order_iter(root)
186
+ 1 2 4 5 3 6 7
185
187
"""
186
188
if not isinstance (node , TreeNode ) or not node :
187
189
return
@@ -210,8 +212,8 @@ def in_order_iter(node: TreeNode) -> None:
210
212
>>> root.left, root.right = tree_node2, tree_node3
211
213
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
212
214
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
213
- >>> in_order_iter(root).strip()
214
- 4 2 5 1 6 3 7
215
+ >>> in_order_iter(root)
216
+ 4 2 5 1 6 3 7
215
217
"""
216
218
if not isinstance (node , TreeNode ) or not node :
217
219
return
@@ -238,8 +240,8 @@ def post_order_iter(node: TreeNode) -> None:
238
240
>>> root.left, root.right = tree_node2, tree_node3
239
241
>>> tree_node2.left, tree_node2.right = tree_node4 , tree_node5
240
242
>>> tree_node3.left, tree_node3.right = tree_node6 , tree_node7
241
- >>> post_order_iter(root).strip()
242
- 4 5 2 6 7 3 1
243
+ >>> post_order_iter(root)
244
+ 4 5 2 6 7 3 1
243
245
"""
244
246
if not isinstance (node , TreeNode ) or not node :
245
247
return
0 commit comments