From 803a77a1df6a251f04d30a303a00d73001e7f0aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Vl=C4=8Dek?= <tomas.vlcek@student.tuke.sk>
Date: Wed, 15 Apr 2026 10:42:48 +0000
Subject: [PATCH] Aktualizovat du5/program.c

---
 du5/program.c | 17 +++++++----------
 1 file changed, 7 insertions(+), 10 deletions(-)

diff --git a/du5/program.c b/du5/program.c
index bb005b1..b42bb3f 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -130,21 +130,18 @@ int main()
     //zisk. korena
     int root = buildATree();
 
-    if (tree[root].yes == -1 && tree[root].no == -1)
-    {
-        printf("Koniec vstupu.\n");
-        return 0;
-    }
-
     printf("Expert z bufetu to vie.\n");
-    if (answerCount != 0) //vypise vysledky/statistiky, AK nebola chyba pri citani odpovedi
+    if (answerCount == 0)
     {
-        printf("Pozna %d druhov ovocia a zeleniny.\n", answerCount);
-        printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+        printf("Chybna databaza\n");
     }
     else
     {
-        printf("Chybna␣databaza");
+        printf("Expert z bufetu to vie.\n");
+        printf("Pozna %d druhov ovocia a zeleniny.\n", answerCount);
+        printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+
+        goThroughTheTree(root);
     }
     
     goThroughTheTree(root);