Python程序打印左子树中的节点
当需要打印左子树中的节点时,可以创建一个类,其中包含可以定义的方法用于设置根节点,执行中序遍历,在根节点的右侧和左侧插入元素等。创建类的实例,可以使用方法执行所需操作。
下面是相同的演示-
示例
class BinaryTree_struct:
def __init__(self, data=None):
self.key = data
self.left = None
self.right = None
def set_root(self, data):
self.key = data
def inorder_traversal(self):
if self.left is not None:
self.left.inorder_traversal()
print(self.key, end=' ')
if self.right is not None:
self.right.inorder_traversal()
def insert_at_left(self, new_node):
self.left = new_node
def insert_at_right(self, new_node):
self.right = new_node
def search_elem(self, key):
if self.key == key:
return self
if self.left is not None:
temp = self.left.search_elem(key)
if temp is not None:
return temp
if self.right is not None:
temp = self.right.search_elem(key)
return temp
return None
def print_left_part(self):
if self.left is not None:
self.left.inorder_traversal()
my_instance = None
print('菜单(假设没有重复的键)')
print('在根节点处插入')
print('在的左侧插入')
print('在的右侧插入')
print('左侧节点')
print('退出')
while True:
my_input = input('您要执行什么操作?').split()
operation = my_input[0].strip().lower()
if operation == 'insert':
data = int(my_input[1])
new_node = BinaryTree_struct(data)
suboperation = my_input[2].strip().lower()
if suboperation == 'at':
my_instance = new_node
else:
position = my_input[4].strip().lower()
key = int(position)
ref_node = None
if my_instance is not None:
ref_node = my_instance.search_elem(key)
if ref_node is None:
print('没有找到该键')
continue
if suboperation == 'left':
ref_node.insert_at_left(new_node)
elif suboperation == 'right':
ref_node.insert_at_right(new_node)
elif operation == 'left':
print('左子树中的节点为:', end='')
if my_instance is not None:
my_instance.print_left_part()
print()
elif operation == 'quit':
break
输出
菜单(假设没有重复的键)
在根节点处插入<data>
在<data>的左侧插入<data>
在<data>的右侧插入<data>
左侧节点
退出
您要执行什么操作?在根节点插入5
您要执行什么操作?在5的左侧插入6
您要执行什么操作?在5的右侧插入8
您要执行什么操作?left
左子树中的节点为:6
您要执行什么操作?quit
使用quit()或Ctrl-D(即EOF)退出
解释
-
创建了具有所需属性的“BinaryTree_struct”类。
-
它有一个“init”函数,用于将左节点和右节点分配为“None”。
-
定义了一个“set_root”方法,用于设置二叉树的根值。
-
它有一个“insert_at_right”方法,可以将元素添加到树的右节点上。
-
它有一个“insert_at_left”方法,可以将元素添加到树的左节点上。
-
另一个名为“inorder_traversal”的方法,执行中序遍历。
-
定义了一个名为“search_elem”的方法,可用于搜索特定元素。
-
另一个名为“print_left_part”的方法,可以在控制台上仅显示二叉树的左部分。
-
创建了一个实例并将其分配给“None”。
-
获取用户输入,以执行所需操作。
-
根据用户的选择执行操作。•在控制台上显示相关输出。