83 lines
1.7 KiB
Python
83 lines
1.7 KiB
Python
import sys
|
|
|
|
|
|
class Node:
|
|
def __init__(self, text, left = None, right = None):
|
|
self.text = text
|
|
self.left = left
|
|
self.right = right
|
|
|
|
def main():
|
|
print("Expert zo zoo to vie.")
|
|
input_data = sys.stdin.read()
|
|
lines = input_data.splitlines()
|
|
if not lines:
|
|
print("Chybna databaza")
|
|
return
|
|
|
|
|
|
q_counter = 0
|
|
a_counter = 0
|
|
idx = 0
|
|
|
|
def parse_tree():
|
|
nonlocal idx, q_counter, a_counter
|
|
if idx >= len(lines):
|
|
return None
|
|
line = lines[idx]
|
|
idx = idx + 1
|
|
|
|
|
|
if line == "":
|
|
return None
|
|
|
|
|
|
if line.endswith('?'):
|
|
q_counter = q_counter + 1
|
|
left = parse_tree()
|
|
if left is None:
|
|
return None
|
|
right = parse_tree()
|
|
if right is None:
|
|
return None
|
|
return Node(line, left, right)
|
|
else:
|
|
a_counter = a_counter + 1
|
|
return Node(line)
|
|
|
|
root = parse_tree()
|
|
|
|
if root is None or idx >= len(lines) or lines[idx] != "":
|
|
print("Chybna databaza")
|
|
return
|
|
|
|
|
|
print(f"Pocet otazok: {q_counter}")
|
|
print(f"Pocet odpovedi: {a_counter}")
|
|
print("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.")
|
|
|
|
idx += 1
|
|
|
|
|
|
current = root
|
|
while True:
|
|
print(current.text)
|
|
if not current.text.endswith('?'):
|
|
print("Koniec")
|
|
break
|
|
if idx >= len(lines):
|
|
print("Koniec vstupu")
|
|
break
|
|
choice = lines[idx]
|
|
idx = idx + 1
|
|
|
|
if choice == 'a':
|
|
current = current.left
|
|
elif choice == 'n':
|
|
current = current.right
|
|
else:
|
|
print("Nerozumiem")
|
|
break
|
|
|
|
if __name__ == '__main__':
|
|
main() |