du5

du5/program.c

@@ -11,24 +11,31 @@ typedef struct Tree {
     struct Tree *no;
 } Tree;
 
+// читання дерева
 Tree* readTree() {
     char line[SIZE];
 
-    if (!fgets(line, SIZE, stdin)) return NULL;
+    if (fgets(line, SIZE, stdin) == NULL) return NULL;
+
     line[strcspn(line, "\r\n")] = 0;
 
+    if (line[0] == '\0') return NULL;
+
     Tree *node = malloc(sizeof(Tree));
     node->yes = NULL;
     node->no = NULL;
 
     if (line[0] == '*') {
         node->isAnswer = 1;
+
         int i = 1;
         while (line[i] == ' ') i++;
+
         strcpy(node->text, line + i);
     } else {
         node->isAnswer = 0;
         strcpy(node->text, line);
+
         node->yes = readTree();
         node->no = readTree();
     }
@@ -36,38 +43,59 @@ Tree* readTree() {
     return node;
 }
 
-int count(Tree *n) {
+// перевірка пустого рядка
-    if (!n) return 0;
+int EmptyLine() {
-    if (n->isAnswer) return 1;
+    char line[SIZE];
-    return count(n->yes) + count(n->no);
+
+    if (fgets(line, SIZE, stdin) == NULL) return 0;
+
+    return (strcmp(line, "\n") == 0);
 }
 
-void freeTree(Tree *n) {
+// кількість листків
-    if (!n) return;
+int Leaves(Tree *node) {
-    freeTree(n->yes);
+    if (!node) return 0;
-    freeTree(n->no);
+    if (node->isAnswer) return 1;
-    free(n);
+
+    return Leaves(node->yes) + Leaves(node->no);
 }
 
-void start(Tree *n) {
+// free
-    char in[SIZE];
+void freeTree(Tree *node) {
+    if (!node) return;
 
-    while (n && !n->isAnswer) {
+    freeTree(node->yes);
-        printf("%s\n", n->text);
+    freeTree(node->no);
+    free(node);
+}
 
-        if (!fgets(in, SIZE, stdin)) return;
+// запуск
+void start(Tree *node) {
+    char input[SIZE];
 
-        int i = 0;
+    while (node && !node->isAnswer) {
-        while (in[i] == ' ' || in[i] == '\t' || in[i] == '\r') i++;
 
-        if (in[i] == 'a')
+        printf("%s\n", node->text);
-            n = n->yes;
+
-        else
+        if (fgets(input, SIZE, stdin) == NULL) {
-            n = n->no;
+            printf("Koniec vstupu\n");
+            return;
+        }
+
+        if (input[0] == 'a') {
+            node = node->yes;
+        }
+        else if (input[0] == 'n') {
+            node = node->no;
+        }
+        else {
+            printf("Nerozumiem\n");
+            return;
+        }
     }
 
-    if (n) {
+    if (node && node->isAnswer) {
-        printf("*%s\n", n->text);
+        printf("*%s\n", node->text);
         printf("Koniec\n");
     }
 }
@@ -82,7 +110,13 @@ int main() {
         return 0;
     }
 
-    printf("Pozna %d druhov ovocia a zeleniny.\n", count(root));
+    if (!EmptyLine()) {
+        printf("Chyba nacitania\n");
+        freeTree(root);
+        return 0;
+    }
+
+    printf("Pozna %d druhov ovocia a zeleniny.\n", Leaves(root));
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
 
     start(root);