From bc6e5952275c12de9e77a8f0ca366de969f5ea00 Mon Sep 17 00:00:00 2001
From: Marege <dutk.nazar@gmail.com>
Date: Tue, 3 Nov 2026 15:35:05 +0100
Subject: [PATCH] fix3u8

---
 u8/program.py | 64 +++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 47 insertions(+), 17 deletions(-)

diff --git a/u8/program.py b/u8/program.py
index 4c6a4af..ef2a401 100644
--- a/u8/program.py
+++ b/u8/program.py
@@ -1,5 +1,42 @@
 import sys
 
+def main():
+    print("Expert zo zoo to vie.")
+    tree_lines = read_tree_lines()
+    if not controla_batabaz(tree_lines):
+        return
+    q = pocet_otazok(tree_lines)
+    a = pocet_odpovedi(tree_lines)
+    tree = build_tree(tree_lines)
+    print("Pocet otazok:", q)
+    print("Pocet odpovedi:", a)
+    print("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.") 
+    koren = tree
+    while koren is not None:
+        if koren["otazka"].endswith("?"):
+            print(koren["otazka"])
+            odpoved = input().rstrip("\r")
+            if odpoved == "a":
+                koren = koren["a"]
+            elif odpoved == "n":
+                koren = koren["n"]
+            else:
+                print("Nerozumiem")
+                return
+        else:
+            print(koren["otazka"])
+            print("Koniec")
+            break
+
+def build_tree(tree_lines):
+    if len(tree_lines) == 0:
+        return None
+    line = tree_lines.pop(0)
+    node = {"otazka": line, "a": None, "n": None}
+    if line.endswith("?"):
+        node["a"] = build_tree(tree_lines)
+        node["n"] = build_tree(tree_lines)
+    return node
 
 def pocet_otazok(tree_lines):
     a = 0
@@ -7,8 +44,6 @@ def pocet_otazok(tree_lines):
         if line.endswith("?"):
             a += 1
     return a
-
-
 def pocet_odpovedi(tree_lines):
     a = 0
     for line in tree_lines:
@@ -16,20 +51,6 @@ def pocet_odpovedi(tree_lines):
             a += 1
     return a
 
-
-def main():
-    print("Expert zo zoo to vie.")
-    tree_lines = read_tree_lines()
-    if tree_lines is None or len(tree_lines) == 0:
-        print("Chybna databaza")
-        return
-    print("Pocet otazok:", pocet_otazok(tree_lines))
-    print("Pocet odpovedi:", pocet_odpovedi(tree_lines))
-    print("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.")
-    print(tree_lines[0])
-    print("Koniec")
-
-
 def read_tree_lines():
     tree_lines = []
     text = input().rstrip("\r")
@@ -38,5 +59,14 @@ def read_tree_lines():
         text = input().rstrip("\r")
     return tree_lines
 
+def controla_batabaz(tree_lines):
+    if tree_lines is None or len(tree_lines) == 0:
+        print("Chybna databaza")
+        return False
+    return True
+
+
+
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()
+