Update cv7/program.c

cv7/program.c

@@ -2,96 +2,121 @@
 #include <stdlib.h>
 #include <string.h>
 
-#define MAX_LINE_LENGTH 100
+#define SIZE 256
 
-// Структура для вузла дерева
-typedef struct Node {
-    char *question;   // Питання
-    char *answer;     // Відповідь (товар)
-    struct Node *yes; // Далі по "так"
-    struct Node *no;  // Далі по "ні"
-} Node;
+// Структура вузла дерева
+struct TreeNode {
+    char *value;            // текст питання або відповіді
+    struct TreeNode *left;   // "так" гілка
+    struct TreeNode *right;  // "ні" гілка
+};
 
-// Функція для створення нового вузла
-Node* create_node(char *question, char *answer) {
-    Node *node = (Node*)malloc(sizeof(Node));
-    node->question = question;
-    node->answer = answer;
-    node->yes = NULL;
-    node->no = NULL;
+// Функція для створення вузла дерева з текстом
+struct TreeNode* createNode(const char *text) {
+    struct TreeNode *node = (struct TreeNode*) malloc(sizeof(struct TreeNode));
+    node->value = strdup(text);
+    node->left = NULL;
+    node->right = NULL;
     return node;
 }
 
-// Функція для зчитування дерева з файлу
-Node* build_tree(FILE *file) {
-    char line[MAX_LINE_LENGTH];
-    if (fgets(line, sizeof(line), file) == NULL) return NULL;
+// Функція для завантаження бази знань у форматі pre-order
+struct TreeNode* loadTree(FILE *file) {
+    char buffer[SIZE];
+    if (!fgets(buffer, SIZE, file) || buffer[0] == '\n') {
+        return NULL;
+    }
 
-    // Видалити пробіли та символи нового рядка в кінці
-    line[strcspn(line, "\n")] = 0;
+    // Створюємо вузол для поточного рядка
+    struct TreeNode *node = createNode(buffer);
 
-    if (line[0] == '*') {  // Це відповідь
-        return create_node(NULL, strdup(line + 1)); // Пропускаємо зірочку
-    } else {  // Це питання
-        Node *node = create_node(strdup(line), NULL);
-        node->yes = build_tree(file);  // Рекурсія для 'так'
-        node->no = build_tree(file);   // Рекурсія для 'ні'
+    // Якщо це відповідь (починається з '*'), то повертаємо листовий вузол
+    if (buffer[0] == '*') {
         return node;
     }
+
+    // Інакше рекурсивно завантажуємо лівий і правий нащадки
+    node->left = loadTree(file);
+    node->right = loadTree(file);
+    return node;
 }
 
-// Функція для підрахунку кількості товарів (листків дерева)
-int count_items(Node *node) {
-    if (node == NULL) return 0;
-    if (node->answer != NULL) return 1; // Листок дерева
-    return count_items(node->yes) + count_items(node->no);
+// Рекурсивна функція для підрахунку товарів у базі знань
+int countItems(struct TreeNode *node) {
+    if (node == NULL) {
+        return 0;
+    }
+    if (node->left == NULL && node->right == NULL) {
+        return 1; // Листовий вузол є товаром
+    }
+    return countItems(node->left) + countItems(node->right);
 }
 
-// Функція для взаємодії з користувачем
-void ask_questions(Node *node) {
-    while (node != NULL) {
-        if (node->answer != NULL) {  // Якщо це відповідь
-            printf("Expert z bufetu to vie.\n");
-            printf("Pozna %d druhov ovocia a zeleniny.\n", count_items(node));
+// Функція для запуску експертної системи з питаннями і відповідями
+void runExpertSystem(struct TreeNode *node) {
+    if (node == NULL) return;
+
+    // Виводимо поточне питання або відповідь
+    printf("%s", node->value);
+
+    // Якщо це листовий вузол, то закінчуємо роботу
+    if (node->left == NULL && node->right == NULL) {
         printf("Koniec\n");
         return;
     }
-        printf("%s\n", node->question);
-        printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
 
-        char response;
-        scanf(" %c", &response); // Читання відповіді
+    // Отримуємо відповідь користувача
+    char answer;
+    scanf(" %c", &answer);
 
-        if (response == 'a') {
-            node = node->yes;
-        } else if (response == 'n') {
-            node = node->no;
+    // Переходимо до відповідного нащадка
+    if (answer == 'a') {
+        runExpertSystem(node->left);
+    } else if (answer == 'n') {
+        runExpertSystem(node->right);
     } else {
-            printf("Nespravny vstup. Koniec.\n");
-            return;
+        printf("Neplatna odpoved. Koniec programu.\n");
     }
 }
+
+// Функція для очищення пам'яті
+void freeTree(struct TreeNode *node) {
+    if (node == NULL) return;
+    freeTree(node->left);
+    freeTree(node->right);
+    free(node->value);
+    free(node);
 }
 
 int main() {
+    // Відкриваємо файл з базою знань
     FILE *file = fopen("data.txt", "r");
     if (file == NULL) {
-        printf("Помилка при зчитуванні файлу.\n");
+        printf("Не вдалося відкрити файл бази знань.\n");
         return 1;
     }
 
-    Node *tree = build_tree(file);
+    // Завантажуємо базу знань в дерево
+    struct TreeNode *root = loadTree(file);
     fclose(file);
 
-    if (tree == NULL) {
-        printf("Помилка в структурі файлу.\n");
+    // Перевірка наявності порожнього рядка після бази знань
+    if (root == NULL) {
+        printf("Помилка: База знань некоректна або пуста.\n");
         return 1;
     }
 
+    // Рахуємо кількість товарів і виводимо повідомлення
+    int itemCount = countItems(root);
     printf("Expert z bufetu to vie.\n");
-    printf("Pozna %d druhov ovocia a zeleniny.\n", count_items(tree));
+    printf("Pozna %d druhov ovocia a zeleniny.\n", itemCount);
+    printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
 
-    ask_questions(tree);
+    // Запуск експертної системи
+    runExpertSystem(root);
+
+    // Звільняємо пам'ять
+    freeTree(root);
 
     return 0;
 }