เมื่อจำเป็นต้องค้นหาจำนวนโหนดที่ไม่ใช่ลีฟในทรี คลาส 'Tree_structure' จะถูกสร้างขึ้น วิธีการตั้งค่ารูท และเพิ่มค่าอื่นๆ จะได้รับการกำหนด มีตัวเลือกต่างๆ ให้ผู้ใช้เลือกได้ ขึ้นอยู่กับทางเลือกของผู้ใช้ การดำเนินการจะดำเนินการกับองค์ประกอบแบบทรี
ด้านล่างนี้เป็นการสาธิตสิ่งเดียวกัน -
ตัวอย่าง
class Tree_structure:
def __init__(self, data=None):
self.key = data
self.children = []
def set_root(self, data):
self.key = data
def add_vals(self, node):
self.children.append(node)
def search_val(self, key):
if self.key == key:
return self
for child in self.children:
temp = child.search_val(key)
if temp is not None:
return temp
return None
def count_non_leaf_node(self):
nonleaf_count = 0
if self.children != []:
nonleaf_count = 1
for child in self.children:
nonleaf_count = nonleaf_count + child.count_non_leaf_node()
return nonleaf_count
tree = None
print('Menu (this assumes no duplicate keys)')
print('add <data> at root')
print('add <data> below <data>')
print('count')
print('quit')
while True:
my_input = input('What operation would you like to perform ? ').split()
operation = my_input[0].strip().lower()
if operation == 'add':
data = int(my_input[1])
newNode = Tree_structure(data)
suboperation = my_input[2].strip().lower()
if suboperation == 'at':
tree = newNode
elif suboperation == 'below':
position = my_input[3].strip().lower()
key = int(position)
ref_node = None
if tree is not None:
ref_node = tree.search_val(key)
if ref_node is None:
print('No such key.')
continue
ref_node.add_vals(newNode)
elif operation == 'count':
if tree is None:
print('The tree is empty ')
else:
count = tree.count_non_leaf_node()
print('The number of non-leaf nodes are : {}'.format(count))
elif operation == 'quit':
break ผลลัพธ์
Menu (this assumes no duplicate keys) add <data> at root add <data> below <data> count quit What operation would you like to perform ? add 34 at root What operation would you like to perform ? add 78 below 34 What operation would you like to perform ? add 56 below 78 What operation would you like to perform ? add 90 below 56 What operation would you like to perform ? count The number of non-leaf nodes are : 3 What operation would you like to perform ? quit
คำอธิบาย
-
สร้างคลาส "Tree_structure" แล้ว
-
มันตั้งค่า 'คีย์' เป็น True และตั้งค่ารายการว่างให้กับลูกของ tree
-
มีฟังก์ชัน 'set_root' ที่ช่วยตั้งค่ารูทสำหรับต้นไม้
-
มีการกำหนดเมธอดชื่อ 'add_vals' ซึ่งช่วยเพิ่มองค์ประกอบให้กับทรี
-
มีการกำหนดวิธีการอื่นที่เรียกว่า 'search_val' ซึ่งช่วยในการค้นหาองค์ประกอบในทรี
-
มีการกำหนดวิธีการอื่นที่ชื่อว่า 'count_non_leaf_nodes' ซึ่งช่วยให้สามารถนับโหนดที่ไม่ใช่ลีฟของต้นไม้ได้
-
เป็นฟังก์ชันแบบเรียกซ้ำ
-
มีสี่ตัวเลือกให้เลือก เช่น 'เพิ่มที่รูท', 'เพิ่มด้านล่าง', 'นับ' และ 'ออก'
-
ขึ้นอยู่กับตัวเลือกที่กำหนดโดยผู้ใช้ การดำเนินการตามลำดับจะดำเนินการ
-
เอาต์พุตนี้จะแสดงบนคอนโซล
