du6.6

du6/program.c

@@ -83,12 +83,16 @@ void freeTree(Node *x) {
 // hlavná funkcia
 int main() {
     Node *root = readTree();          // načítame strom zo vstupu
+     printf("Expert z bufetu to vie.\n"); // vždy vypíšeme tento riadok
 
+    if (!root) {                    // ak strom je prázdny alebo chybne načítaný
+        printf("Chybna databaza\n");
+        return 0;
+    }    
     fgets(s, MAX, stdin);             // prečítame prázdny riadok medzi databázou a vstupom používateľa
 
     int k = count(root);              // spočítame počet listov (druhov tovaru)
      
-    printf("Expert z bufetu to vie.\n");       
     printf("Pozna %d druhov ovocia a zeleniny.\n", k);
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");