Tree의 리프 노드 수를 계산해야 하는 경우 'Tree_structure' 클래스가 생성되고 루트 값을 추가하는 메소드 및 기타 자식 값이 정의됩니다. 사용자가 선택할 수 있는 다양한 옵션이 제공됩니다. 사용자의 선택에 따라 Tree 요소에 대해 작업을 수행합니다.
아래는 동일한 데모입니다 -
예시
class Tree_structure:
def __init__(self, data=None):
self.key = data
self.children = []
def set_root_node(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(key)
if temp is not None:
return temp
return None
def count_leaf_node(self):
leaf_nodes = []
self.count_leaf_node_helper_fun(leaf_nodes)
return len(leaf_nodes)
def count_leaf_node_helper_fun(self, leaf_nodes):
if self.children == []:
leaf_nodes.append(self)
else:
for child in self.children:
child.count_leaf_node_helper_fun(leaf_nodes)
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)
sub_op = my_input[2].strip().lower()
if sub_op == 'at':
tree = newNode
elif sub_op == 'below':
my_pos = my_input[3].strip().lower()
key = int(my_pos)
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_leaf_node()
print('The number of 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 78 at root What operation would you like to perform ? add 90 below 78 What operation would you like to perform ? add 8 below 78 What operation would you like to perform ? count The number of leaf nodes are : 2 What operation would you like to perform ? quit
설명
-
'Tree_structure' 클래스가 생성됩니다.
-
'key'를 True로 설정하고 빈 목록을 트리의 자식으로 설정합니다.
-
트리의 루트 값을 설정하는 데 도움이 되는 'set_root' 기능이 있습니다.
-
트리에 요소를 추가하는 데 도움이 되는 'add_vals'라는 메서드가 정의되어 있습니다.
-
트리에서 요소를 검색하는 데 도움이 되는 'search_val'이라는 또 다른 메서드가 정의되어 있습니다.
-
'count_leaf_nodes'라는 또 다른 메서드가 정의되어 트리의 리프 노드 수를 얻는 데 도움이 됩니다.
-
이전에 정의한 함수를 호출하는 'count_leaf_nodes_helper'라는 또 다른 메서드가 정의되어 있습니다. 이것은 재귀 함수입니다.
-
'루트에 추가', '아래에 추가', '카운트' 및 '종료'와 같은 4가지 옵션이 제공됩니다.
-
사용자가 부여한 옵션에 따라 각각의 작업을 수행합니다.
-
이 출력은 콘솔에 표시됩니다.