From 5aa8f2e9382041063dbe48dd7524b21b8c462b0c Mon Sep 17 00:00:00 2001
From: Anton Dolozin <anton.dolozin@student.tuke.sk>
Date: Sun, 9 Nov 2025 17:26:09 +0100
Subject: [PATCH] Fixed leaf count

---
 du6/program.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/du6/program.c b/du6/program.c
index a9c0dc7..80436d8 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -93,6 +93,10 @@ int main(void){
     int numOfLeaves = count_leaves(tree);
     char answer[SIZE];
     printf("Expert z bufetu to vie.\n");
+    if(!tree){
+        printf("Chybna databaza\n");
+        return 0;
+    }
     printf("Pozna %d druhov ovocia a zeleniny.\n", numOfLeaves);
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.");
     while(tree){