From 14a335d999780cb4f2e36a1f6d6cbbfc67bbfa1d Mon Sep 17 00:00:00 2001
From: Illia Korpan <illia.korpan@student.tuke.sk>
Date: Wed, 15 Apr 2026 09:26:10 +0200
Subject: [PATCH] 123

---
 du5/program.c | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/du5/program.c b/du5/program.c
index 8574274..7e217bb 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -117,10 +117,31 @@ int main(void) {
     printf("Pozna %d druhov ovocia a zeleniny.\n", leaves);
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
 
-    if (root && !root->is_answer) {
-        printf("%s\n", root->text);
+    Node *cur = root;
+    char buf[128];
+    while (cur) {
+        if (cur->is_answer) {
+            printf("*%s\n", cur->text);
+            break;
+        }
+        printf("%s\n", cur->text);
+        if (!fgets(buf, sizeof(buf), stdin)) {
+            printf("Nerozumiem\n");
+            return 0;
+        }
+        char *p = buf;
+        while (*p && isspace((unsigned char)*p)) p++;
+        char c = *p;
+        if (c == 'a' || c == 'A') {
+            cur = cur->yes;
+        } else if (c == 'n' || c == 'N') {
+            cur = cur->no;
+        } else {
+            printf("Nerozumiem\n");
+            return 0;
+        }
     }
-    printf("Koniec vstupu\n");
+    printf("Koniec\n");
 
     free_tree(root);
     for (int i = 0; i < lines_count; ++i) free(lines[i]);