Merge pull request #80 from gupta2140/master

Fixed infinite loop while entering, and preorder traversal.
This commit is contained in:
Harshil 2017-06-07 16:33:32 +05:30 committed by GitHub
commit ef9082b468

View File

@ -1,7 +1,7 @@
"""
This is pure python implementation of tree traversal algorithms
"""
from __future__ import print_function
import queue
@ -25,22 +25,25 @@ def build_tree():
node_found = q.get()
print("Enter the left node of %s: " % node_found.data, end="")
left_data = eval(input())
if left_data >= 0:
if left_data < 0:
return tree_node
elif left_data >= 0:
left_node = TreeNode(left_data)
node_found.left = left_node
q.put(left_node)
print("Enter the right node of %s: " % node_found.data, end="")
right_data = eval(input())
if right_data >= 0:
if right_data < 0:
return tree_node
elif right_data >= 0:
right_node = TreeNode(right_data)
node_found.right = right_node
q.put(right_node)
return tree_node
def pre_order(node):
if not isinstance(node, TreeNode) or not node:
print("Invalid input")
#print("Invalid input")
return
print(node.data, end=" ")
pre_order(node.left)